Low-Light Image Object Detection Based on Improved YOLOv5 Algorithm

被引:4
|
作者
Shu Ziting [1 ,2 ]
Zhang Zebin [1 ,2 ]
Song Yaozhe [1 ,2 ]
Wu Mengmeng [1 ,2 ]
Yuan Xiaobing [1 ]
机构
[1] Chinese Acad Sci, Shanghai Inst Microsyst & Informat Technol, Key Lab Microsyst Technol, Shanghai 201800, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
关键词
object detection; low-light image; YOLOv5; network; image enhancement; attention mechanism;
D O I
10.3788/LOP212965
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
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.
引用
收藏
页数:8
相关论文
共 19 条
  • [1] Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934]
  • [2] Chen W., 2021, arXiv preprint
  • [3] The Pascal Visual Object Classes (VOC) Challenge
    Everingham, Mark
    Van Gool, Luc
    Williams, Christopher K. I.
    Winn, John
    Zisserman, Andrew
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 88 (02) : 303 - 338
  • [4] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [5] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [6] Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement
    Guo, Chunle
    Li, Chongyi
    Guo, Jichang
    Loy, Chen Change
    Hou, Junhui
    Kwong, Sam
    Cong, Runmin
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 1777 - 1786
  • [7] LIME: Low-Light Image Enhancement via Illumination Map Estimation
    Guo, Xiaojie
    Li, Yu
    Ling, Haibin
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (02) : 982 - 993
  • [8] EnlightenGAN: Deep Light Enhancement Without Paired Supervision
    Jiang, Yifan
    Gong, Xinyu
    Liu, Ding
    Cheng, Yu
    Fang, Chen
    Shen, Xiaohui
    Yang, Jianchao
    Zhou, Pan
    Wang, Zhangyang
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 : 2340 - 2349
  • [9] Selective Kernel Networks
    Li, Xiang
    Wang, Wenhai
    Hu, Xiaolin
    Yang, Jian
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 510 - 519
  • [10] Receptive Field Block Net for Accurate and Fast Object Detection
    Liu, Songtao
    Huang, Di
    Wang, Yunhong
    [J]. COMPUTER VISION - ECCV 2018, PT XI, 2018, 11215 : 404 - 419