基于改进YOLOv5的复杂场景多目标检测

来源：电子测量技术杂志2022年第23期北京时间：

作者：强 栋，王占刚

单位：北京信息科技大学信息与通信工程学院 北京 100101

摘要：针对多目标图像检测环境复杂、目标物位置数据冗余且长宽高数据大小不一的问题，利用神经网络算法可以有效提高不同类目标物并行检测的准确度和稳定性，提出一种基于改进YOLOv5网络的多目标检测方法。首先依据不同目标物的空间尺度大小，改进模型的特征融合方法，添加多尺度特征检测层以减小多目标检测时的误差，同时增加自适应特征增强模块（Adaptive Feature Adjustment），降低网络的误检率与漏检率；然后使用 K-means++ 算法估计候选框，获得更优的框参数；最后在损失函数中使用EIOU（Efficient IOU Loss）做优化。实验表明：改进后的方法mAP（mean average precision）达到76.48%，相比经典YOLOv5网络提升了3.2%，小尺寸目标物检测准确度均值增加6.3%。改进方法网络延续YOLOv5网络的轻量高效，对于多尺度目标物检测获得更优的检测精度，能够实现更准确的实时多目标检测。

关键词：神经网络;多目标检测; YOLOv5; 自适应特征增强;损失函数优化

基金资助：国家重点研发计划课题资助（编号：2018YFC1800203）；北京市科技创新服务能力建设-基本科研业务费(市级)(科研类)(PXM2019_014224_000026)