全局与局部图像特征自适应融合的小目标检测算法
CSTR:
作者:
作者单位:

1. 西安建筑科技大学 信息与控制工程学院,西安 710055;2. 陕西省岩土与地下空间工程重点实验室,西安 710055

作者简介:

通讯作者:

E-mail: zhaoliang@xauat.edu.cn.

中图分类号:

TP183

基金项目:

国家自然科学基金项目(51209167,12002251);陕西省自然科学基金项目(2019JM-474);西安市科技计划项目(2020KJRC0055);陕西省岩土与地下空间工程重点实验室开放基金项目(YT202004).


Small object detection algorithm based on adaptive fusion of global and local image features
Author:
Affiliation:

1. College of Information and Control Engineering,Xián University of Architecture and Technology,Xián 710055,China;2. Shaanxi Provincial Key Laboratory of Geotechnical and Underground Space Engineering,Xián 710055,China

Fund Project:

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

    针对现有目标检测算法对于小目标检测精度低的问题,提出一种全局与局部图像特征自适应融合的一阶段小目标检测算法SODet.首先,将Transformer与卷积神经网络相结合构建主干网络,分别提取图像全局和局部信息,并利用自适应特征选择模块AFS对二者输出进行融合;然后,在特征融合网络中利用额外尺度特征图进行特征融合,同时利用大目标抑制单元约束大目标特征表达、转移小目标特征,输出4个尺度的特征图送入预测网络;最后,在损失函数部分针对小目标检测利用EIOU和Focal loss进行优化.实验结果表明,SODet算法在MS COCO验证集上$\rm AP_S$达到31.5%,相比于其他算法具有较强的竞争力,同时具有较高的推理速度.

    Abstract:

    Aiming at the problem that the existing object detectors have low accuracy for small objects. A one-stage small object detector (SODet) is proposed to adaptively fuse global and local image features. Firstly, the Transformer and the convolutional neural network (CNN) are combined to construct a backbone network to extract global and local information of the image respectively. Then the adaptive feature selection (AFS) module is used to fuse the outputs of the Transformer and CNN. Then, extra-scale feature maps are adopted in the feature fusion network. At the same time, the large object restraint unit is applied to constrain the expression of large object features and transfer small object features. The feature maps of four scales are sent to the prediction network. Finally, in the loss function, the EIOU and Focal loss are used to optimize small object detection. The experimental results show that the SODet has 31.5% in terms of $\rm AP_S$ on the MS COCO verification set, which is more competitive than other algorithms and has a higher inference speed.

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

赵亮,刘世鹏.全局与局部图像特征自适应融合的小目标检测算法[J].控制与决策,2023,38(4):935-943

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