摘要
YOLOv3算法在目标检测和识别方面具有检测精度高、速度快等优点,但其损失函数中高宽比部分的损失运算对大目标与小目标的区分度不明显,导致损失计算结果精度较差,小目标预测框尺寸不准确。为此,对YOLOv3算法的损失函数进行了改进,将宽高坐标误差部分修改为与目标宽高真实值相关的形式,并将原模型的非极大值抑制函数的有重叠区域的检测框的分数由固定值改为衰减函数形式,最后对改进模型进行了车辆检测试验。研究表明,改进模型在不影响车辆检测速度的同时,精确率P、召回率R、平均准确度mAP和调和F1值分别提高了0.105、0.1、0.021、0.102,改进模型的整体性能优于原YOLOv3模型。
- 单位