基于轻量化YOLOv4的交通信息实时检测方法

来源：计算机应用杂志2023年第1期北京时间：

作者：郭克友, 李雪, 杨民

单位：北京工商大学 人工智能学院，北京 100048

摘要：针对日常道路场景下的车辆目标检测问题，提出一种轻量化的YOLOv4交通信息实时检测方法。首先，制作了一个多场景、多时段的车辆目标数据集，并利用K-means++算法对数据集进行预处理；其次，提出轻量化YOLOv4检测模型，利用MobileNet?v3替换YOLOv4的主干网络，降低模型的参数量，并引入深度可分离卷积代替原网络中的标准卷积；最后，结合标签平滑和退火余弦算法，使用LeakyReLU激活函数代替MobileNet?v3浅层网络中原有的激活函数，从而优化模型的收敛效果。实验结果表明，轻量化YOLOv4的权值文件为56.4 MB，检测速率为85.6 FPS，检测精度为93.35%，表明所提方法可以为实际道路中的交通实时信息检测及其应用提供参考。

关键词：目标检测,深度学习,图像处理,轻量化,YOLOv4