Research on Improved YOLOv5 for Low-Light Environment Object Detection

Abstract

Object detection in low-light scenarios has been widely acknowledged as a significant research area in the field of computer vision, presenting a challenging task. Aiming at the low detection accuracy of mainstream single-stage object detection models in low-light scenarios, this paper proposes a detection model called DK_YOLOv5 based on YOLOv5, specifically designed for such scenarios. First, a low-light image enhancement algorithm with better results is selected to generate enhanced images that achieve relatively better visual effects and amplify target features. Second, the SPPF layer is improved to an R-SPPF module with faster inference speed and stronger feature expression ability. Next, we replace the C3 module with the C2f module and incorporate an attention mechanism to develop the C2f_SKA module, enabling richer gradient information flow and reducing the impact of noise features. Finally, the model detection head is replaced with a decoupled head suitable for the object detection task in this scenario to improve model performance. Additionally, we expand the Exdark dataset to include low-light data of underground mine scenario targets, named Mine_Exdark. Experimental results demonstrate that the proposed DK_YOLOv5 model achieves higher detection accuracy than other models in low-light scenarios, with an mAP0.5 of 71.9% on the Mine_Exdark dataset, which is 4.4% higher than that of YOLOv5.