摘要
本申请涉及一种基于YOLOv4的道路车辆检测系统及方法,所述道路车辆检测系统包括道路车辆数据采集模块、道路车辆检测模型和终端设备,所述道路车辆检测模型为改进的YOLOv4算法模型,所述改进的YOLOv4算法模型的主干网络以MobileVit网络作为主干特征提取网络,颈部Neck包括空间金字塔池化层SPP和预测层PANet;模型的定位损失函数为CDIOU损失函数。本发明采用GridMask图像增强方法进行图片预处理,并通过训练好的道路车辆检测模型进行道路车辆检测,改善了现有的YOLOv4模型精度低,检测速度慢,不够轻量型的问题,在提升检测效率的同时,保证了模型的检测精度。