来源：期刊VIP网 时间：
作者：文靖杰;王勇;李金龙;张渝;
单位：西南交通大学物理科学与技术学院;
摘要：文中提出一种融合多头注意力机制、ROIAlign和Soft-NMS的FasterR-CNN目标检测算法，旨在解决原始Faster R-CNN目标检测网络中存在的检测精度低、漏检、误检的问题。首先，为了提高Faster R-CNN的感知能力，提取特征图中的重要特征并降低对无关特征的提取，在网络中嵌入注意力机制；接着，针对共享全连接层的降维操作导致的一些区域的细节信息被忽略，造成局部信息的丢失，采用一维卷积代替共享全连接层实现权重计算的任务，以捕捉更广泛的空间信息；然后为了提供更丰富的特征表达能力，在注意力机制中引入多头机制分别对特征的不同部分进行重要性的加权；为了减少在特征提取时原图信息的丢失，使用ROI Align替换ROI Pooling算法；最后，在算法后处理中引入Soft-NMS替换传统非极大抑制（NMS）算法以减少漏检和误检情况。实验证明，改进后的Faster R-CNN目标检测网络对感兴趣目标的定位能力得到提高，漏检和误检情况减少，平均检测精度得到显著提升。
关键词：机器视觉;;目标检测;;Faster R-CNN;;ROI Align;;多头注意力机制;;Soft-NMS
基金资助：自然基金重点国际（地区）合作与交流项目（61960206010）;; 四川省科技计划项目（2021YJ0080）
