AODs-CLYOLO: An Object Detection Method Integrating Fog Removal and Detection in Haze Environments
-
Published:2024-08-20
Issue:16
Volume:14
Page:7357
-
ISSN:2076-3417
-
Container-title:Applied Sciences
-
language:en
-
Short-container-title:Applied Sciences
Author:
Liang Xinyu1ORCID, Liang Zhengyou1ORCID, Li Linke1, Chen Jiahong1
Affiliation:
1. School of Computer and Electronic Information, Guangxi University, Nanning 530004, China
Abstract
Foggy and hazy weather conditions can significantly reduce the clarity of images captured by cameras, making it difficult for object detection algorithms to accurately recognize targets. This degradation can cause failures in autonomous or assisted driving systems, posing severe safety threats to both drivers and passengers. To address the issue of decreased detection accuracy in foggy weather, we propose an object detection algorithm specifically designed for such environments, named AODs-CLYOLO. To effectively handle images affected by fog, we introduce an image dehazing model, AODs, which is more suitable for detection tasks. This model incorporates a Channel–Pixel (CP) attention mechanism and a new Contrastive Regularization (CR), enhancing the dehazing effect while preserving the integrity of image information. For the detection network component, we propose a learnable Cross-Stage Partial Connection Module (CSPCM++), which is used before the detection head. Alongside this, we integrate the LSKNet selective attention mechanism to improve the extraction of effective features from large objects. Additionally, we apply the FocalGIoU loss function to enhance the model’s performance in scenarios characterized by sample imbalance or a high proportion of difficult samples. Experimental results demonstrate that the AODs-CLYOLO detection algorithm achieves up to a 10.1% improvement in the mAP (0.5:0.95) metric compared to the baseline model YOLOv5s.
Funder
National Natural Science Foundation of China
Reference33 articles.
1. Bijelic, M., Gruber, T., Mannan, F., Kraus, F., Ritter, W., Dietmayer, K., and Heide, F. (2020, January 13–19). Seeing Through Fog Without Seeing Fog: Deep Multimodal Sensor Fusion in Unseen Adverse Weather. Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA. 2. Redmon, J., and Farhadi, A. (2017, January 21–26). YOLO9000: Better, Faster, Stronger. Proceedings of the 30th IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA. 3. Redmon, J., and Farhadi, A. (2018). YOLOv3: An Incremental Improvement. arXiv. 4. Bochnovsniy, A., Wang, C.Y., and Liao, H.Y.M. (2020). YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv. 5. Wang, C.Y., Bochkovskiy, A., and Liao, H.Y. (2023, January 18–22). YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, BC, Canada.
|
|