Research on Low-Light Environment Object Detection Algorithm Based on YOLO_GD
Jian Li,Xin Wang,Qi Chang,Yongshan Wang,Haifeng Chen
DOI: https://doi.org/10.3390/electronics13173527
IF: 2.9
2024-09-06
Electronics
Abstract:In low-light environments, the presence of numerous small, dense, and occluded objects challenges the effectiveness of conventional object detection methods, failing to achieve desirable results. To address this, this paper proposes an efficient object detection network, YOLO_GD, which is designed for precise detection of targets in low-light scenarios. This algorithm, based on the foundational framework of YOLOv5s, implements a cross-layer feature fusion method founded on an information gathering and distribution mechanism. This method mitigates the issue of information loss during inter-layer feature exchange and, building on this, constructs a Bi-level routing spatial attention module to reduce computational redundancy caused by the self-attention mechanism, thereby enhancing the model's detection accuracy for small objects. Furthermore, through the introduction of a novel deformable convolution, a cross-stage local feature fusion module is established, enabling the model to capture the complex features of input data more accurately and improve detection precision for dense objects. Lastly, the introduction of a probabilistic distance metric in the bounding box regression loss function enhances the network model's generalization capability, further increasing detection accuracy in occluded scenarios. Experimental results on the ExDark dataset demonstrate that compared to YOLOv5, there is a 5.97% improvement in mean average precision (mAP), effectively enhancing object detection performance in low-light conditions.
engineering, electrical & electronic,computer science, information systems,physics, applied