Object detection algorithm based on attention mechanism in foggy weather

Wanye Gu; Yuecheng Yu; Liming Cai; Jinlong Shi; Yongzheng Li; Shixin Huang

doi:10.1117/12.2682368

26 May 2023 Object detection algorithm based on attention mechanism in foggy weather

Wanye Gu, Yuecheng Yu, Liming Cai, Jinlong Shi, Yongzheng Li, Shixin Huang

Proceedings Volume 12700, International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023); 1270026 (2023) https://doi.org/10.1117/12.2682368
Event: International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023), 2023, Nanchang, China

Abstract

In this paper, we propose a foggy weather object detection model based on an attention mechanism, to address the problem of low detection accuracy, missed detection and false detection when general object detection models are applied directly to foggy scenes. Firstly, to enhance the detection network's multi-scale expression ability and sensitivity to the target, a residual module that integrates the attention mechanism replaces the BottleNeck module of the backbone network. This design improves the network's ability to extract features and locate targets at a fine-grained level. Secondly, the CIOU loss function replaces the original loss function, improving the stability of the bounding box regression process. Thirdly, the K-means++ clustering algorithm is used to generate anchors suitable for the dataset in this paper. Furthermore, the object detection dataset in foggy scenes is further enriched based on the atmospheric scattering model. Experimental results indicate that the proposed method's mAP in light fog, medium fog and dense fog scenes is increased by 7.4%, 6.05% and 6.36%, respectively, compared to the original YOLOv5s. This improvement in accuracy significantly reduces the missed detection rate and false detection rate, effectively enhancing object detection performance in foggy weather.

Citation Download Citation

Wanye Gu, Yuecheng Yu, Liming Cai, Jinlong Shi, Yongzheng Li, and Shixin Huang "Object detection algorithm based on attention mechanism in foggy weather", Proc. SPIE 12700, International Conference on Electronic Information Engineering and Data Processing (EIEDP 2023), 1270026 (26 May 2023); https://doi.org/10.1117/12.2682368

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available