Complex dark environment-oriented object detection method based on YOLO-AS

被引:0
作者
Ren, Bin [1 ]
Xu, Zhaohui [1 ]
Zhao, Junwu [1 ]
Hao, Rujiang [1 ]
Zhang, Jianchao [1 ]
机构
[1] Shijiazhuang Tiedao Univ, Sch Mech Engn, Shijiazhuang 050043, Hebei, Peoples R China
基金
中国国家自然科学基金;
关键词
Dark environment; Object detection; Image enhancement; Attention mechanism;
D O I
10.1038/s41598-025-07348-0
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
For object detection in complex dark environments, the existing methods generally have problems such as low detection accuracy, false detection and missed detection. These problems lead to a lack of key information and incomplete context information, which seriously affects the detection effect. Therefore, this paper proposes an object detection method for complex dark environment based on YOLO-AS. First, the Zero-DCES image enhancement module is designed to improve the image quality of the dark environment via adaptive contrast enhancement. Second, a YOLO-AS detection model is constructed, which integrates ECA_ASPP and the SK attention mechanism. The receptive field is expanded by dilated convolution, and dynamic feature detection is realized by combining channel attention, which effectively enhances the ability of multiscale feature expression. Finally, the model is tested on the ExDark dataset and the LOL dataset, and compared with current mainstream models such as YOLOv5s, YOLOv8n and YOLOv11. Experiments show that the proposed method achieves 78.39% map@50 on the ExDark dataset while maintaining a basically unchanged detection speed, which is 5.78% higher than that of the benchmark model, and significantly improves the detection accuracy in complex dark environments.
引用
收藏
页数:14
相关论文
共 49 条
[1]  
Agarap A F., Deep Learning using Rectified Linear Units, DOI DOI 10.48550/ARXIV.1803.08375
[2]   Zero-DCE xT: A Computational Approach to Addressing Low-Light Image Enhancement Challenges [J].
Ahmadi, Candra ;
Chen, Jiann-Liang ;
Hsiao, Yu-Yi .
IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS II-EXPRESS BRIEFS, 2024, 71 (09) :4401-4405
[3]   Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement [J].
Cai, Yuanhao ;
Bian, Hao ;
Lin, Jing ;
Wang, Haoqian ;
Timofte, Radu ;
Zhang, Yulun .
2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, :12470-12479
[4]   End-to-End Object Detection with Transformers [J].
Carion, Nicolas ;
Massa, Francisco ;
Synnaeve, Gabriel ;
Usunier, Nicolas ;
Kirillov, Alexander ;
Zagoruyko, Sergey .
COMPUTER VISION - ECCV 2020, PT I, 2020, 12346 :213-229
[5]   Automatic Contrast-Limited Adaptive Histogram Equalization With Dual Gamma Correction [J].
Chang, Yakun ;
Jung, Cheolkon ;
Ke, Peng ;
Song, Hyoseob ;
Hwang, Jungmee .
IEEE ACCESS, 2018, 6 :11782-11792
[6]   Instance Segmentation in the Dark [J].
Chen, Linwei ;
Fu, Ying ;
Wei, Kaixuan ;
Zheng, Dezhi ;
Heide, Felix .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2023, 131 (08) :2198-2218
[7]   FeatEnHancer: Enhancing Hierarchical Features for Object Detection and Beyond Under Low-Light Vision [J].
Hashmi, Khurram Azeem ;
Kallempudi, Goutham ;
Stricker, Didier ;
Afzal, Muhammamd Zeshan .
2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION, ICCV, 2023, :6702-6712
[8]  
Hou QB, 2021, Arxiv, DOI [arXiv:2103.02907, DOI 10.48550/ARXIV.2103.02907, 10.48550/arXiv.2103.02907]
[9]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
[10]   PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer [J].
Jiang, Wentao ;
Liu, Si ;
Gao, Chen ;
Cao, Jie ;
He, Ran ;
Feng, Jiashi ;
Yan, Shuicheng .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :5193-5201