改进的YOLOv3目标检测算法

作者：曹春键; 臧强^*; 王泽嘉; 屠壮

来源：中国科技论文, 2021, 16(11): 1195-1201.

• 目标检测
• YOLOv3算法
• 特征融合
• 非极大值抑制
• 平均精度均值

针对YOLOv3目标检测算法存在网络参数量大、检测精度不够高的问题,首先,使用一种轻量化YOLOv3特征提取网络的方法,降低特征提取网络部分的参数量;其次,提出一种多级特征融合网络结构,提高YOLOv3算法特征层的检测效果;最后,采用一种软化的非极大值抑制(soft non-maximun suppression, Soft-NMS)算法,在检测阶段有效避免重叠目标下的漏检。结果表明,相比于YOLOv3算法,改进YOLOv3算法的参数量降低了46%,模型大小约为原模型的50%,在PASCAL VOC 2007数据集上的平均精度均值(mean average precision, mAP)提升了3.5%。

• 出版日期2021
• 单位南京信息工程大学; 自动化学院