Zeng Qingliang, Zhou Guangyu, Wan Lirong, Wang Liang, Xuan Guantao, Shao Yuanyuan
College of Mechanical and Electronic Engineering, Shandong University of Science and Technology, Qingdao 266590, China.
College of Mechanical and Electrical Engineering, Shandong Agricultural University, Taian 271018, China.
Sensors (Basel). 2024 Feb 15;24(4):1246. doi: 10.3390/s24041246.
To address the lightweight and real-time issues of coal sorting detection, an intelligent detection method for coal and gangue, Our-v8, was proposed based on improved YOLOv8. Images of coal and gangue with different densities under two diverse lighting environments were collected. Then the Laplacian image enhancement algorithm was proposed to improve the training data quality, sharpening contours and boosting feature extraction; the CBAM attention mechanism was introduced to prioritize crucial features, enhancing more accurate feature extraction ability; and the EIOU loss function was added to refine box regression, further improving detection accuracy. The experimental results showed that Our-v8 for detecting coal and gangue in a halogen lamp lighting environment achieved excellent performance with a mean average precision (mAP) of 99.5%, was lightweight with FLOPs of 29.7, Param of 12.8, and a size of only 22.1 MB. Additionally, Our-v8 can provide accurate location information for coal and gangue, making it ideal for real-time coal sorting applications.
为解决煤炭分选检测的轻量化和实时性问题,基于改进的YOLOv8提出了一种煤矸石智能检测方法Our-v8。采集了在两种不同光照环境下不同密度的煤和矸石图像。然后提出拉普拉斯图像增强算法来提高训练数据质量,锐化轮廓并增强特征提取;引入CBAM注意力机制对关键特征进行优先级排序,增强更准确的特征提取能力;添加EIOU损失函数来优化边界框回归,进一步提高检测精度。实验结果表明,Our-v8在卤素灯照明环境下检测煤和矸石具有优异的性能,平均精度均值(mAP)为99.5%,轻量化程度高,浮点运算次数(FLOPs)为29.7,参数(Param)为12.8,大小仅为22.1MB。此外,Our-v8可以为煤和矸石提供准确的位置信息,使其非常适合实时煤炭分选应用。