关键词:
目标检测
密集交通场景
EIOU
CBAM
K-means++
摘要:
道路交通场景下目标检测任务往往由于物体相互遮挡、远处目标像素低等原因,导致检测精度低,出现漏检、误检等问题,为此提出一种改进YOLOv5s的交通场景密集目标检测模型YOLOv5-ECFK。首先,将边界框回归损失函数由CIOU替换为EIOU,提高模型评估预测框的准确性;其次,在主体网络结构中融合CBAM注意力机制,帮助网络在具有密集对象场景中找到注意力区域;然后,通过增加小尺度特征融合层和检测层,提升模型小目标检测精度;最后,使用K-means++算法重新聚类锚框,使生成的锚框更适合实验数据集,提升算法整体性能。实验数据集由公开数据集Rope3D和自制数据集联合组成。实验结果表明,最终改进算法YOLOv5-ECFK相比于YOLOv5s、mAP@0.5/%和mAP@0.5:0.95/%分别提升3.4%、2.7%,Precision和Recall分别提升3.5%、1.3%,最终改进算法对道路交通场景下的密集目标具有优秀的检测效果。