高级检索

基于LMIENet图像增强的矿井下低光环境目标检测方法

LMIENet enhanced object detection method for low light environment inunderground mines

  • 摘要: 煤矿井下工作环境复杂,存在人造光源亮度低、粉尘多和水气密度大等不利因素,导致现有的目标检测算法在应用到煤矿井下时,存在提取特征困难、目标识别和定位精度低等问题。提出一种煤矿井下低照度环境目标检测算法,由矿井低光图像增强模块LMIENet和目标检测模块组成,使用图像增强模块对原始图像进行画质提升,恢复各类图像信息,再使用目标检测网络对增强图像进行特定目标检测,有效提高检测的精确度。在图像增强模块中,改进Zero-DCE算法设计轻量级增强参数预测网络,计算像素级增强参数矩阵,用于低光照图像的亮度调整和画质增强,该网络通过设计的非参考损失函数隐性衡量图像的增强效果,引导网络进行无监督学习,使网络能够不依赖配对数据集对原始图像进行自适应的画质增强。目标检测模块中,采用YOLO v8n目标检测模型,其轻量化的模型尺寸和高灵活性可避免模型整体复杂度过高;采用Focal-EIoU Loss改进回归损失函数,有效加速模型收敛并提升模型检测精度。实验结果显示:与经典目标检测算法Faster R–CNN,SSD, RetinaNet, FCOS等相比,提出算法在自建矿井人员数据集上表现出色,低光照环境下目标检测的mAP@0.5达到98.0%,mAP@0.5∶0.95达64.8%,在实验环境中单帧图像推理时间仅11 ms,优于其他对比方法,证明提出算法能够有效实现在煤矿井下低照度复杂环境下的目标检测,且耗时短、计算效率高。

     

    Abstract: The underground working environment of coal mines is complex, with unfavorable factors such as low brightness of artificial light sources, high dust content, and high water vapor density. This leads to difficulties in extracting features and low accuracy of object recognition and positioning when existing object detection algorithms are applied to coal mines. An object detection algorithm for low illumination environments in coal mines is proposed, which consists of an low-light mine image enhancement module LMIENet and a object detection module. The image enhancement module is used to improve the image quality of the original image, restore various image information, and then use a target detection network to perform specific target detection on the enhanced image, effectively improving the accuracy of detection. In the image enhancement module, a lightweight enhancement parameter prediction network is designed with reference to the zero reference depth curve estimation algorithm, and the pixel level enhancement parameter matrix is calculated for image quality enhancement and brightness adjustment of low light images. The network implicitly measures the image enhancement effect through the designed non-reference loss function, and guides the network to conduct unsupervised learning, Enable the network to adaptively enhance the image quality of the original image without relying on paired datasets. In the object detection module, the YOLO v8n object detection model is adopted, which has a lightweight model size and high flexibility to avoid excessive overall model complexity; Using Focal EIoU Loss to improve regression loss, accelerate model convergence, and improve model detection accuracy. The experimental results show that compared with classic object detection algorithms such as Faster R–CNN, SSD, RetinaNet, etc., the proposed algorithm performs well on the self-made coal mine object detection dataset, and is effective in object detection in low light environments mAP@0.5 reaches 98.0%, mAP@0.5∶0.95 reaches 64.8%, and the running time of a single frame image in the experimental environment is only 11 ms, which is superior to other comparison methods. It is proven that the proposed algorithm can effectively achieve object detection in low illumination and complex environments in coal mines, with short time consumption and high computational efficiency.

     

/

返回文章
返回