[口头报告]An improved YOLOX model for accurate foreign object detection in coal mining industry based on deep learning algorithm
An improved YOLOX model for accurate foreign object detection in coal mining industry based on deep learning algorithm
编号:76
稿件编号:269 访问权限:仅限参会人
更新:2024-05-16 19:01:54
浏览:620次
口头报告
报告开始:2024年05月30日 16:40 (Asia/Shanghai)
报告时间:15min
所在会议:[S6] Clean Processing, Conversion and Utilization of Energy Resources » [S6-1] Afternoon of May 30th
暂无文件
摘要
Accurate real-time detection of foreign object during industrial coal mining and processing is crucial to ensuring equipment and personnel safety, as well as maintaining the quality of coal product. The complexity of real industrial coal mining and processing environment poses challenges to vison-based foreign object detection. This work proposes a SHA-DH-YOLOX algorithm to enhance detection accuracy. The proposed algorithm stands out with three improvements. First, Shuffle Attention mechanism (SHA) is integrated into the YOLOX backbone to strengthen the feature extraction of essential information from input images. Second, Dynamic head (Dyhead) is introduced after feature fusion to enhance the detector's representation capability for scale-, spatial-, and task-awareness. Third, the original Intersection over Union (IoU) loss function is replaced by SCYLLA-IoU (SIoU) for more accurate bounding boxes and enhanced training stability. These improvements work collaboratively with YOLOX-M, making the proposed algorithm outperform state-of-the-art baseline models including YOLOv5, YOLOv6, YOLOv7, and original YOLOX series. The developed SHA-DH-YOLOX algorithm improves AP50 by 1.87 - 7.97% compared to baseline models of equivalent size. Robustness tests further demonstrate the stability of SHA-DH-YOLOX model in facing diverse harsh scenarios. This pioneering work provides valuable tools to achieve safe and reliable coal mining and processing.
关键字
foreign object detection,YOLOX,shuffle attention mechanism,dynamic head,SIoU
发表评论