基于改进YOLOv5的车辆检测方法研究

分享到

【获取途径】超星期刊网
【作者】赵月爱，王哲
【刊名】现代计算机
【作者单位】太原师范学院计算机科学与技术学院
【年份】2024
【卷号】第8期
【ISSN】1007-1423
【关键词】YOLOv5s 车辆检测 Mish BiFPN EIoU
【摘要】基于YOLOv5s对交通道路上前方车辆的检测进行了研究，为提高车辆检测精度，对原YOLOv5s网络模型进行了改进。首先，将Mish激活函数应用于YOLOv5s模型之中代替原有的ReLU 函数，使用更平滑的Mish激活函数有效避免梯度消失的问题。其次，采用BiFPN 作为特征金字塔，增加了特征传递的信息通道，提升模型的感知能力和上下文信息的关联能力。最后，引入EIoU 作为损失函数的一部分，准确地表示预测框和真实框之间的位置关系，间接地使收敛速度提升。在D-City数据集上的实验表明，改进后的YO...
【文献类型】期刊