Laser & Optoelectronics Progress, Volume. 60, Issue 4, 0404001(2023)
Low-Light Image Object Detection Based on Improved YOLOv5 Algorithm
Aiming at the low detection accuracy of existing object detection algorithms in a low-light environment, a dual-channel low-light image object detection algorithm called YOLOv5_DC according to an enhanced YOLOv5 algorithm is suggested. First, we synthesize low-light images using Gamma transformation and superimposing Gaussian noise to expand the dataset and promote the network's generalization. Second, a feature enhancement module is proposed. The channel attention method is used to integrate the low-level characteristics of the improved image and the original image to decrease the effect of noisy features and increase the network's feature extraction capabilities. Finally, a feature location module is added to the neck network to boost the response value of the feature map in the target area, allowing the network to focus more on the target area and improve the network detection capabilities. The experimental results show that the proposed YOLOv5_DC algorithm achieves higher detection accuracy. On the low-light object detection dataset known as ExDark*, the mean average precision (mAP) @0.5 of the proposed algorithm reaches 71.85%, which is 1.28 percentage points higher than the original YOLOv5 algorithm.
Get Citation
Copy Citation Text
Ziting Shu, Zebin Zhang, Yaozhe Song, Mengmeng Wu, Xiaobing Yuan. Low-Light Image Object Detection Based on Improved YOLOv5 Algorithm[J]. Laser & Optoelectronics Progress, 2023, 60(4): 0404001
Category: Detectors
Received: Nov. 16, 2021
Accepted: Dec. 24, 2021
Published Online: Feb. 14, 2023
The Author Email: Yuan Xiaobing (yuanxb@mail.sim.ac.cn)