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

摘要

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