基于改进YOLOv3算法在道路目标检测中的应用

• 作者: 谭芳喜 肖世德 周亮君 李晟尧
• 来源期刊: 计算机技术与发展
• 关键词: 目标检测 YOLOv3 深度可分离卷积 SENet DIoU-NMS算法
• 年，卷（期）: 2021,（8）
• 所属期刊栏目: 应用前沿与综合
• 页码范围: 118-123
• 页数: 6页
• 分类号: TP391
• 语种: 中文
• DOI: 10.3969/j.issn.1673-629X.2021.08.020

摘要

为了提高道路环境下目标检测的准确率和实时性,提出一种基于YOLOv3的改进检测算法.通过深度可分离卷积模块减少模型计算量,提高模型的实时性;采用K-Means++聚类算法代替原来的K-Means算法生成本数据集所需的先验锚点框,解决K-Means算法受初始点选取的影响较大,聚类结果不稳定的问题;在YOLOv3的多尺度预测网络中引入SENet(squeeze-and-excitation networks),加强网络对特征的学习能力;改进位置损失函数,解决使用IoU(intersection over union)度量时存在无法反映预测框与真实框重合度大小、无法优化IoU为零等问题;利用DIoU-NMS(基于Distance-IoU的非极大值抑制)去除冗余框,减少错误抑制,提高检测精度.实验结果表明,改进算法相对于原算法在检测耗时降低的同时,对5类常见目标检测的准确率均有提升.