AW‐YOLO: A Multi‐Object Detection Network for Autonomous Driving Under all Weather Conditions
ABSTRACT Over the years, object detection technology based on deep learning has attracted extensive research in autonomous driving. Achieving a robust object detection network under all weather conditions (e.g., sunny, fog, nighttime, rain and snow) is highly significant for autonomous driving systems, which ensure safety by recognising pedestrians, vehicles, traffic lights, etc. This paper proposes a robust multi‐object detection network named All Weather‐You Only Look Once (AW‐YOLO) based on YOLOv8, with a trade‐off between precision and lightweightness. Considering the blurring or absence of the salient object features of the image under all weather conditions, we propose a developed dilation‐wise residual (D‐DWR) module. Specifically, it combines the dilatation‐wise residual module with the dilated re‐param block using a large kernel convolution to see wide without going deep, greatly improving the feature extraction ability. Moreover, we introduce an efficient dynamic upsampler (DySample) that formulates upsampling from the viewpoint of point sampling and avoids dynamic convolution, which can improve the network's ability to feature fusion. Lightweight is an essential requirement for autonomous driving. To this end, we adopt a multi‐scale shared detection head (MSSD‐Head) to achieve lightweight deployment in autonomous vehicles. Experimental results show that the mAP50‐95 values of AW‐YOLO on the KITTI and Adverse Conditions Dataset with Correspondences (ACDC) datasets exceed the baseline model YOLOv8 by 1.7% and 1.5%, respectively. Meanwhile, the parameters and model size of AW‐YOLO have decreased by 21.4% and 20.4%, respectively.
Authors
- Yihuan Zhang (ORCID: https://orcid.org/0000-0001-6341-4057)
- Jiaji Pan (ORCID: https://orcid.org/0000-0001-8954-0780)
- Miaolei He (ORCID: https://orcid.org/0000-0003-4743-2663)
- Lu Dai (ORCID: https://orcid.org/0000-0001-9795-8968)
- Dingping Chen
- Siming Huang (ORCID: https://orcid.org/0000-0002-0912-1082)
- Qian Gu
- Xiangdi Yue (ORCID: https://orcid.org/0000-0001-9453-7981)
Institutions
- China Aerospace Science and Industry Corporation (China) (CN)
- Hunan Normal University (CN)
Publication Details
- Journal
- IET Image Processing
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1049/ipr2.70111
- Citations
- 1
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.28
Funders
- Hunan Provincial Innovation Foundation for Postgraduate