基于注意力掩模融合的目标检测算法

• 作者: 何小海 卿粼波 吴晓红 滕奇志 董潇潇
• 来源期刊: 液晶与显示
• 学科: 工学
• 关键词: 计算机视觉 目标检测 注意力掩模 特征金字塔 多尺度检测
• 年，卷（期）: 2019,（8）
• 所属期刊栏目: 图像处理
• 页码范围: 825-833
• 页数: 9页
• 分类号: TP391
• 字数: 3558字
• 语种: 中文
• DOI: 10.3788/YJYXS20193408.0825

摘要

在计算机视觉任务中,平衡目标检测的精度与速度对于后续的实际应用如目标跟踪和识别起到关键作用.基于此,提出了一种基于注意力掩模融合的目标检测算法.首先,通过VGG网络提取特征,经过初步二分类和回归后得到一系列预选框;然后,将上述预选框输入到特征金字塔结构中,通过构建注意力掩模模块自适应地学习有效特征,同时融合特征金字塔结构与注意力掩模模块得到更具表征性的特征;最后经过多分类和回归得到多尺度的检测结果.在PASCAL VOC 2007和PASCAL VOC 2012数据集上展开了实验,测试集结果显示,在交集并集比(IOU)为0.5的条件下,对于320×320的图片输入,平均精度均值(mAP)分别为81.0％ 和79.0％,检测速度为60.9 fps.本文算法将注意力信息结合到目标检测中,实现了通用目标检测的精度和速度均衡.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	何小海	四川大学电子信息学院	395	2334	21.0	30.0
2	滕奇志	四川大学电子信息学院	198	900	14.0	21.0
3	吴晓红	四川大学电子信息学院	141	571	11.0	16.0
4	卿粼波	四川大学电子信息学院	181	565	11.0	15.0
5	董潇潇	四川大学电子信息学院	1	3	1.0	1.0