科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对复杂交通场景下自动驾驶汽车对遮挡目标以及小目标漏检情况严重、精度较低的问题，本文提出改进的YOLOv4目标检测方法。首先提出一种新的非极大值抑制算法Soft-DIoU-NMS以提升算法对遮挡目标的精确定位能力，其次改进K-means聚类算法生成更准确的候选框，最后引入焦点损失缓解样本之间的不均衡问题。实验结果表明，改进后的YOLOv4检测精度达到89.91%，检测速度达到35.52 f/s，能够很好地解决复杂交通场景的目标检测问题。
-
单位广东交通职业技术学院; 大连理工大学; 经济管理学院
