结合注意力机制的深度学习图像目标检测

• 作者: 孙萍 张永军 胡旭东
• 来源期刊: 计算机工程与应用
• 学科: 工学
• 关键词: 目标检测 卷积神经网络(CNN) 注意力机制 特征降维
• 年，卷（期）: 2019,（17）
• 所属期刊栏目: 模式识别与人工智能
• 页码范围: 180-184
• 页数: 5页
• 分类号: TP389.1
• 字数: 4642字
• 语种: 中文
• DOI: 10.3778/j.issn.1002-8331.1902-0155

摘要

利用卷积神经网络进行目标检测时,提取的卷积特征具有很强的平移不变性,这将削弱模型的定位性能.事实上,目标对象通常具有不同的子区域特征和宽高比特性,但在目前流行的两阶段目标检测框架中,很少考虑这些具有平移尺度敏感性的特征成分.为了优化模型的特征表达,将在两阶段目标检测框架中引入与子区域特征和宽高比特性相关的注意力特征库,并生成注意力特征图对原始的ROI池化特征进行优化.另外,在注意力特征图的辅助下,模型特征维度可以有效地进行缩减.实验结果表明,引入注意力模块后,模型的检测精度和检测速度有明显提升.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	张永军	武汉大学遥感信息工程学院	55	688	15.0	25.0
2	孙萍	武汉大学遥感信息工程学院	4	4	1.0	2.0