可重叠抑制的Anchor-free目标检测算法

• 作者: 汤毅 王坤阳 张宽 滕国伟
• 来源期刊: 电视技术
• 学科: 工学
• 关键词: Anchor-free 重叠抑制 目标检测算法 鲁棒性
• 年，卷（期）: 2020,（9）
• 所属期刊栏目: 监管与检测
• 页码范围: 66-72
• 页数: 7页
• 分类号: TP391
• 语种: 中文
• DOI: 10.16280/j.videoe.2020.09.015

摘要

针对Anchor-base目标检测算法的预定义边界框尺度、长宽比固定和需要预先设计的缺陷,提出了一种新的重叠抑制Anchor-free目标检测算法.该算法舍弃了预定义边界框生成,选择直接回归预测特征层上像素点到对象边界框四边的距离,摆脱了预定义边界框的尺度和长宽比限制,可以实现对任意形状的目标对象的识别和边界框回归.在边界框回归时,将周围重叠遮挡对象的影响考虑在内,在边界框回归损失函数中增加与重叠对象边界框之间距离成反比的辅助损失函数,使对象的预測边界框尽可能远离周围重叠遮挡对象的预测边界框和边界框真实值.实验结果表明,提出的Anchor-free目标检测算法能够显著提升极端形状对象和重叠遮挡对象的检测鲁棒性.