基于注意力机制和多层次特征融合的目标检测算法
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

湖南省自然科学基金资助项目(2021JJ50049);湖南省教育厅科学研究基金资助重点项目(21A0607)


Target Detection Algorithm Based on Attention Mechanism and Multi-Level Feature Fusion
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了提高目标检测的准确率,提出一种基于注意力机制和多层次特征融合的图像目标检测算法。该算法在Cascade R-CNN模型的基础上,以RseNet50为主干网络,通过嵌入简单的注意力模块(SAM)来提高网络的判别能力;其次,利用深度可分离卷积改进特征金字塔网络(FPN),设计了多层次特征融合模块(MFFM),对多尺度特征进行融合,以丰富特征图的信息量,并对不同层次的特征图赋予相应的权重以平衡不同尺度的特征信息;最后,结合目标检测方法中的区域建议网络(RPN)结构获取目标的候选区域进行分类和回归处理,确定检测目标的位置和类别。实验结果表明,相较于Cascade R-CNN目标检测算法,该算法的检测精度提升了约2.0%。

    Abstract:

    For an improvement of the accuracy of object detection, an image object detection algorithm has thus been proposed based on attention mechanism and multiple feature fusion. On the basis of the Cascade R-CNN model, the algorithm uses RseNet50 as the backbone network, with a simple attention module (SAM) embedded so as to improve the discrimination ability of the network. Secondly, a multi-level feature fusion module (MFFM) is designed by using the deep separable convolution to improve the feature pyramid network (FPN), followed by a fusion of the multi-scale features to enrich the information of feature maps, with the corresponding weights given to the feature maps of different levels to balance the feature information of different scales. Finally, combined with the region proposal network (RPN) structure in the target detection method, the candidate regions of the target can be obtained for classification and regression processing to determine the location and category of the detection target. Experimental results show that compared with Cascade R-CNN target detection algorithm, the detection accuracy of the proposed algorithm has been improved by approximately 2.0%.

    参考文献
    相似文献
    引证文献
引用本文

周秋艳,肖满生,范双南.基于注意力机制和多层次特征融合的目标检测算法[J].湖南工业大学学报,2023,37(1):61-68.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-03-16
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-01-03
  • 出版日期: