融合自适应注意力机制的Faster R-CNN目标检测算法 |
| |
引用本文: | 王启胜,王凤随,陈金刚,刘芙蓉.融合自适应注意力机制的Faster R-CNN目标检测算法[J].激光与光电子学进展,2022(12):391-396. |
| |
作者姓名: | 王启胜 王凤随 陈金刚 刘芙蓉 |
| |
作者单位: | 1. 安徽工程大学电气工程学院;2. 检测技术与节能装置安徽省重点实验室;3. 高端装备先进感知与智能控制教育部重点实验室 |
| |
基金项目: | 安徽省自然科学基金(2108085MF197,1708085MF154);;安徽高校省级自然科学研究重点项目(KJ2019A0162); |
| |
摘 要: | 针对Faster R-CNN目标检测算法存在的定位和检测精度问题,设计了一种可嵌入Faster R-CNN目标检测算法并进行端到端训练的可移动的注意力(MA)模型。首先,为了获取更加精确的空间位置信息,MA采用两个自适应最大池化分别基于输入特征图的水平和竖直两个方向进行特征聚合,生成两个独立的方向感知特征图;其次,为了防止模型过拟合,使用Sigmoid激活函数增加网络非线性;最后,为了充分利用已经得到的空间位置信息,将具有非线性的两个特征图与输入特征图依次相乘以增强输入特征图的表征能力。实验结果表明:基于MA改进的Faster R-CNN目标检测算法有效地提升了网络对感兴趣目标的定位能力,并且平均检测精度也得到了明显的提升。
|
关 键 词: | 机器视觉 目标检测 Faster R-CNN 注意力机制 卷积神经网络 ResNet-50 |
|