一种改进的非极大值抑制算法在YOLOv3模型中的应用研究

• 作者: 陈鹏程 戚宝仁 孔德宇
• 来源期刊: 数字技术与应用
• 关键词: 目标检测 YOLOv3算法 非极大值抑制
• 年，卷（期）: 2021,（1）
• 所属期刊栏目: 算法分析
• 页码范围: 121-123
• 页数: 3页
• 分类号: TP391
• 语种: 中文
• DOI: 10.19695/j.cnki.cn12-1369.2021.01.37

摘要

YOLOv3目标检测模型在进行候选区域确定时大多使用非极大值抑制算法,这类算法在多目标检测任务可能出现重复检测等问题,针对这一问题,本文在前人研究的soft-NMS算法基础上进行优化,提出I-NMS算法,并将该算法应用到YOLOv3算法中进行建模,通过实验对比验证表明,I-NMS算法有助于解决YOLOv3算法中的重复检测问题.