Wang Rui, Zhao He, Xu Zhengwei, Ding Yaming, Li Guowei, Zhang Yuxin, Li Hua
School of Computer Science and Technology, Changchun University of Science and Technology, Changchun, Jilin, China.
Department of Geophysics, Chengdu University of Technology, Chengdu, Sichuan, China.
Front Neurorobot. 2023 Mar 9;17:1058723. doi: 10.3389/fnbot.2023.1058723. eCollection 2023.
As a crucial component of the autonomous driving task, the vehicle target detection algorithm directly impacts driving safety, particularly in inclement weather situations, where the detection precision and speed are significantly decreased. This paper investigated the You Only Look Once (YOLO) algorithm and proposed an enhanced YOLOv4 for real-time target detection in inclement weather conditions. The algorithm uses the Anchor-free approach to tackle the problem of YOLO preset anchor frame and poor fit. It better adapts to the detected target size, making it suitable for multi-scale target identification. The improved FPN network transmits feature maps to unanchored frames to expand the model's sensory field and maximize the utilization of model feature data. Decoupled head detecting head to increase the precision of target category and location prediction. The experimental dataset BDD-IW was created by extracting specific labeled photos from the BDD100K dataset and fogging some of them to test the proposed method's practical implications in terms of detection precision and speed in Inclement weather conditions. The proposed method is compared to advanced target detection algorithms in this dataset. Experimental results indicated that the proposed method achieved a mean average precision of 60.3%, which is 5.8 percentage points higher than the original YOLOv4; the inference speed of the algorithm is enhanced by 4.5 fps compared to the original, reaching a real-time detection speed of 69.44 fps. The robustness test results indicated that the proposed model has considerably improved the capacity to recognize targets in inclement weather conditions and has achieved high precision in real-time detection.
作为自动驾驶任务的关键组成部分,车辆目标检测算法直接影响驾驶安全,尤其是在恶劣天气情况下,此时检测精度和速度会显著降低。本文研究了You Only Look Once(YOLO)算法,并提出了一种增强的YOLOv4,用于在恶劣天气条件下进行实时目标检测。该算法采用无锚点方法来解决YOLO预设锚框和拟合不佳的问题。它能更好地适应检测目标的大小,适用于多尺度目标识别。改进后的特征金字塔网络(FPN)将特征图传输到无锚框,以扩大模型的感知域并最大化模型特征数据的利用率。解耦头检测头提高了目标类别和位置预测的精度。通过从BDD100K数据集中提取特定的带标签照片并对其中一些进行雾化处理,创建了实验数据集BDD-IW,以测试所提出方法在恶劣天气条件下的检测精度和速度方面的实际应用。在该数据集中将所提出的方法与先进的目标检测算法进行了比较。实验结果表明,所提出的方法平均精度均值达到60.3%,比原始的YOLOv4高5.8个百分点;算法的推理速度比原始算法提高了4.5帧每秒,达到了69.44帧每秒的实时检测速度。鲁棒性测试结果表明,所提出的模型在恶劣天气条件下识别目标的能力有了显著提高,并在实时检测中实现了高精度。