基于改进YOLOv4-tiny算法的车距预警方法

Vehicle distance warning method based on improved YOLOv4⁃tiny algorithm

针对现有网络难以实时精确识别道路检测目标及车距的问题,提出了一种基于改进YOLOv4-tiny算法的车距预警方法。首先,总结了YOLOv4-tiny算法的特征提取结构,分析了原网络结构的不足之处。其次,在原网络中增加了空间金字塔池化层SPPF进一步提取目标特征,增强深层次语义信息表达能力,并将特征金字塔网络(FPN)结构添加下采样通道和CSPnet层,充分融合多尺度图像特征,避免浅层信息丢失。最后,使用Mosaic数据增强方法丰富数据集训练样本,并将改进YOLOv4-tiny算法与单目测距原理相结合,依据车距大小设置3种级别的信息提示进行车距预警实验。结果表明:本文算法在PASCAL VOC数据集上的检测速度为43帧/s,平均精度达到81.25%,较YOLOv4-tiny算法提高了3.59%。可见,改进YOLOv4-tiny算法在满足检测实时性要求的同时,具备良好的目标检测精度,对提升车距预警方法的使用效果具有指导意义。

Aiming at the problem that the existing network is difficult to accurately recognize the target of road detection and vehicle distance in real time,a vehicle distance warning method based on improved YOLOv4-tiny algorithm was proposed.Firstly,the feature extraction structure of YOLOv4-tiny algorithm was summarized,and the shortcomings of the original network structure were analyzed.Secondly,SPPF of spatial pyramid pooling layer was added to the original network to further extract target features and enhance the ability to express deep semantic information.Feature Pyramid Network(FPN)structure was added with down-sampling channel and CSPnet layer to fully integrate multi-scale image features and avoid the loss of shallow information.Finally,the Mosaic data enhancement method was used to enrich the data set training samples,and the improved YOLOv4-tiny algorithm was combined with the principle of single visual distance detection.The vehicle distance warning experiment was conducted by setting three levels of information cues according to the vehicle distance.The results show that the detection speed of the proposed algorithm on PASCAL VOC dataset is 43 frames/s,and the average accuracy is 81.25%,which is 3.59%higher than that of YOLOv4-tiny algorithm.It can be seen that the improved YOLOv4-tiny algorithm has good target detection accuracy while meeting the real-time requirements of detection,which has guiding significance for improving the application effect of vehicle distance warning method.