科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种目标检测中使用IoU加强自注意力机制的方法,其特点是计算proposal boxes之间的IoU,使用IoU作为位置信息,使其加强proposal之间的自注意力机制；将该模块嵌入到目标检测网络中,对网络进行训练,实现检测器检测能力的增强。本发明与现有技术相比具有不引入参数,增加极少量的计算量,就能显著地增强检测器在COCO数据集和CrowdHuman数据集上的指标的特点,方法简单,效率高,明显地提高了检测精度,使用IoU作为位置信息对自注意力机制进行加强,较好的解决了当前注意力机制中缺失位置信息的问题。
- 单位
