基于注意力机制的NMS在目标检测中的研究

作     者：张长伦 张翠文 王恒友 何强 刘屹伟 Zhang Changlun;Zhang Cuiwen;Wang Hengyou;He Qiang;Liu Xiwei

作者机构：北京建筑大学理学院北京100044 

基　　金：国家自然科学基金项目(62072024) 北京建筑大学北京未来城市设计高精尖创新中心项目(UDC2017033322,UDC2019033324) 北京建筑大学市属高校基本科研业务费专项(X20084)资助 

出 版 物：《电子测量技术》 (Electronic Measurement Technology)

年 卷 期：2021年第44卷第19期

页      码：82-88页

摘      要：非最大值抑制算法是目标检测中选择定位准确框的主要算法。针对算法仅以分类得分作为标准可能去除得分低但定位准确的预测框,并且对于有遮挡的情况更不友好的情况,提出了A-NMS方法。该方法将注意力机制融入非极大值抑制算法中,利用位置信息与框的得分信息相结合来调整框的最后得分。此外,还提出了改进的基于距离的交并比损失函数,重新定义了损失项,并将其引入到非极大值抑制中代替IOU来计算框间的交并比。最后将两种改进算法融合到3种经典的目标检测中,在Pascal-VOC 2012和MS-COCO 2017数据集上对上述两种算法进行了验证,结果表明检测精度得到了1%~2%的提升。

主 题 词：非最大抑制 目标检测 注意力机制 

学科分类：081203[081203] 08[工学] 0835[0835] 0812[工学-测绘类] 

D　O　I：10.19651/j.cnki.emt.2107024

馆 藏 号：203106397...