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

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

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

engineering of communications and transportation safetytarget detectionYOLOv4-tiny algorithmmonocular distance measurementvehicles warning

王宏志、宋明轩、程超、解东旋

展开 >

长春工业大学 计算机科学与工程学院,长春 130012

交通运输安全工程 目标检测 YOLOv4-tiny算法 单目测距 车距预警

国家自然科学基金项目吉林省教育厅重点项目

61903047JKH20210754KJ

2024

吉林大学学报(工学版)
吉林大学

吉林大学学报(工学版)

CSTPCD北大核心
影响因子:0.792
ISSN:1671-5497
年,卷(期):2024.54(3)
  • 31