|
|||||
|
|
| 基于改进YOLOv5的小目标交通标志检测算法 | |
| 引用本文： | 徐鑫,方凯.基于改进YOLOv5的小目标交通标志检测算法[J].湖北汽车工业学院学报,2023(4):17-21. |
| 作者姓名： | 徐鑫 方凯 |
| 作者单位： | 湖北汽车工业学院电气与信息工程学院 |
| 基金项目： | 湖北省教育厅科学技术研究项目（D20201802）；;教育部产学研合作协同育人项目（202002009056）； |
| 摘 要： | 针对小目标交通标志检测精度低的问题，提出基于改进YOLOv5的小目标交通标志检测算法。优化YOLOv5的主干网络结构，采用微小目标检测层替换原始大目标检测层，调整下采样倍数，将MobileViT Block融入颈部网络，采用离线增强与在线增强的方法对数据集进行处理。实验结果表明：与其他目标检测算法相比，改进算法的小目标交通标志检测均值平均精度为85.3%，参数量降低了39%,FPS为70，满足实时检测的要求。 |
| 关 键 词： | 交通标志检测 YOLOv5 MobileVit Block 小目标检测 |