基于图像增强和改进YOLOv8的煤矿低光照目标检测

Low-light target detection in coal mines based on image enhancement and improved YOLOv8

  • 摘要: 目前煤矿井下图像增强技术在实际应用中存在稳定性不足和生成图像质量波动较大的问题,影响后续目标检测的精度;而目前基于YOLOv8的煤矿井下目标检测技术在低光照环境下应用时,由于图像特征弱化和信息丢失,YOLOv8的性能仍然存在一定局限性。针对上述问题,提出一种基于图像增强和改进YOLOv8的煤矿低光照目标检测算法。采用去噪概率扩散模型(DDPM)对原始图像进行去噪和增强处理,恢复图像的光照及细节信息;在YOLOv8基础上进行改进,通过引入低频滤波增强模块(LEF)和特征增强模块(FEM)提高低光照图像的特征提取性能,并将YOLOv8模型原有的CIoU回归损失更换为MPDIoU,得到YOLOv8−DLFM;使用YOLOv8−DLFM进行目标检测,提高目标检测准确性和鲁棒性。实验结果表明:① 与目前主流的图像增强方法进行对比,DDPM的峰值信噪比为28.379 dB,结构相似性为0.886,感知相似性为0.104,表现出优越的图像重建质量和结构相似性。② YOLOv8−DLFM在综合性能上表现优异,准确率、召回率和mAP@0.5分别达到0.878,0.791和0.896,帧率达到88.6帧/s,相较于原始YOLOv8n模型,YOLOv8−DLFM的准确率、召回率与mAP@0.5分别提升了8.13%,6.6%和8.74%。③ 与目前主流目标检测模型相比,YOLOv8−DLFM在复杂低光照环境下具有更强的鲁棒性和更高的检测精度;在目标遮挡、光照干扰、目标稀疏和目标密集等典型工况下,YOLOv8−DLFM展现出较高的鲁棒性和适应性。

     

    Abstract: At present, the existing image enhancement techniques for underground coal mines suffer from insufficient stability and large fluctuations in the quality of generated images, which affect the accuracy of subsequent target detection. Meanwhile, target detection methods based on YOLOv8 also face certain limitations in low-light environments due to weakened image features and information loss. To address these problems, a low-light target detection algorithm for coal mines based on image enhancement and improved YOLOv8 was proposed. The Denoising Diffusion Probabilistic Model (DDPM) was used to denoise and enhance the original images, restoring illumination and detail information. Based on YOLOv8n, improvements were made by introducing a Low-Frequency Filter Enhancement Module (LEF) and a Feature Enhancement Module (FEM) to enhance feature extraction performance for low-light images. The original CIoU regression loss function in YOLOv8n was replaced with MPDIoU, yielding the YOLOv8-DLFM model. The YOLOv8-DLFM was then used for target detection to improve accuracy and robustness. Experimental results showed that: ① compared with mainstream image enhancement methods, DDPM achieved a peak signal-to-noise ratio of 28.379 dB, a structural similarity index of 0.886, and a perceptual similarity of 0.104, demonstrating superior image reconstruction quality and structural similarity. ② YOLOv8-DLFM exhibited excellent overall performance, with precision, recall, and mAP@0.5 reaching 0.878, 0.791, and 0.896, respectively, and a frame rate of 88.6 frames/s. Compared with the original YOLOv8n model, the precision, recall, and mAP@0.5 of YOLOv8-DLFM increased by 8.13%, 6.6%, and 8.74%, respectively. ③ Compared with mainstream target detection models, YOLOv8-DLFM demonstrated stronger robustness and higher detection accuracy under complex low-light environments. It also exhibited high robustness and adaptability under typical conditions such as target occlusion, lighting interference, sparse targets, and dense targets.

     

/

返回文章
返回