首页|感受野扩展与多分支聚合的目标检测方法

感受野扩展与多分支聚合的目标检测方法

扫码查看
目标检测旨在实现对图像中目标的精确识别和定位,是计算机视觉中一个重要的研究领域.基于深度学习的目标检测已取得长足的发展,但依然存在不足之处.大的下采样系数带来的语义信息有利于图像分类,但下采样过程中不可避免地会造成信息损失,导致模型特征提取不充分,从而检测准确性下降.针对上述问题,提出一种感受野增强与多分支聚合模型用于目标检测.首先,设计感受野增强模块,以扩大主干网络的感受野.该模块可以获取目标上下文线索,且不改变特征的空间分辨率,可以缓解下采样过程中目标信息丢失问题.然后,为了充分利用卷积神经网络的局部性以及自注意力机制的长距离特征依赖特性,构建感受野扩展复合主干网络,以保留局部特征以及提高模型的全局特征感知能力.最后,提出多分支聚合检测头网络,在3个预测分支之间形成信息流动,融合分支之间的特征信息,以提高模型检测能力.在MS COCO数据集上进行了验证实验,结果表明所提模型的平均精度优于多种主流目标检测模型.
Object Detection with Receptive Field Expansion and Multi-branch Aggregation
Object detection aims to achieve accurate recognition and localization of objects in images and is an important research area in computer vision.Deep learning-based object detection has made great progress,but there are still shortcomings.The se-mantic information brought by large down-sampling coefficients is beneficial to image classification,but the down-sampling process inevitably brings information loss,resulting in insufficient model feature extraction and thus a decrease in detection accu-racy.To address these problems,this paper proposes a receptive field enhancement and multi-branch aggregation network for ob-ject detection.First,the receptive field enhancement module is designed to expand the receptive field of the backbone network.This module can acquire object context cues and can alleviate the problem of object information loss during down-sampling be-cause it does not change the feature spatial resolution.Then,in order to take full advantage of the localization of convolutional neural networks and the long-range feature-dependent property of the self-attention mechanism,the receptive field expanding composite backbone network is constructed to retain local features as well as to improve the global feature perception capability of the model.Finally,a multi-branch aggregation detection head network is proposed to form information flow between three predic-tion branches and fuse feature information between branches to improve the detection capability of the model.Validation experi-ments are carried out on MS COCO datasets,and the results show that the average accuracy of the proposed model is better than that of many mainstream object detection models.

Object detectionSelf-attention mechanismReceptive field expansionFeature fusionDecoupled head

阙越、甘梦晗、刘志伟

展开 >

华东交通大学信息工程学院 南昌 330013

目标检测 自注意力机制 感受野扩展 特征融合 解耦检测头

国家自然科学基金江西省自然科学基金

6236203220232BAB212011

2024

计算机科学
重庆西南信息有限公司(原科技部西南信息中心)

计算机科学

CSTPCD北大核心
影响因子:0.944
ISSN:1002-137X
年,卷(期):2024.51(z1)
  • 30