Laser & Optoelectronics Progress, Volume. 60, Issue 4, 0404001(2023)

Low-Light Image Object Detection Based on Improved YOLOv5 Algorithm

Ziting Shu1,2, Zebin Zhang1,2, Yaozhe Song1,2, Mengmeng Wu1,2, and Xiaobing Yuan1、*
Author Affiliations
  • 1Key Laboratory of Microsystem Technology, Shanghai Institute of Microsystem and Information Technology, Chinese Academy of Sciences, Shanghai 201800, China
  • 2University of Chinese Academy of Sciences, Beijing 100049, China
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    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)

    DOI:10.3788/LOP212965

    Topics