Low-Light Image Object Detection Based on Improved YOLOv5 Algorithm

被引:7
作者
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, DOI 10.48550/arXiv.2107.14382
[3]   The Pascal Visual Object Classes (VOC) Challenge [J].
Everingham, Mark ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 88 (02) :303-338
[4]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[5]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[6]   Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement [J].
Guo, Chunle ;
Li, Chongyi ;
Guo, Jichang ;
Loy, Chen Change ;
Hou, Junhui ;
Kwong, Sam ;
Cong, Runmin .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1777-1786
[7]   LIME: Low-Light Image Enhancement via Illumination Map Estimation [J].
Guo, Xiaojie ;
Li, Yu ;
Ling, Haibin .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (02) :982-993
[8]   EnlightenGAN: Deep Light Enhancement Without Paired Supervision [J].
Jiang, Yifan ;
Gong, Xinyu ;
Liu, Ding ;
Cheng, Yu ;
Fang, Chen ;
Shen, Xiaohui ;
Yang, Jianchao ;
Zhou, Pan ;
Wang, Zhangyang .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :2340-2349
[9]   Selective Kernel Networks [J].
Li, Xiang ;
Wang, Wenhai ;
Hu, Xiaolin ;
Yang, Jian .
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 [J].
Liu, Songtao ;
Huang, Di ;
Wang, Yunhong .
COMPUTER VISION - ECCV 2018, PT XI, 2018, 11215 :404-419