基于多尺度特征聚合注意力的无锚目标检测算法

ANCHOR-FREE OBJECT DETECTION ALGORITHM BASED ON MULTI-SCALE FEATURE AGGREGATION ATTENTION NETWORK

  • 摘要: 针对单尺度注意力目标检测方法在复杂场景下效果不佳的问题,在CenterNet的基础上,提出基于多尺度特征聚合注意力的无锚目标检测算法。该算法采用多尺度特征聚合注意力网络(Multi-scale Feature Aggregation Attention Network,MFANet),以实现聚合主干网络不同感受野的特征,同时关注特征图重要区域。此外,对CenterNet的关键点预测分支进行重新设计,使用更深的卷积网络以获得更好的预测。在UAVDT数据集以及COCO数据集上的实验结果表明,保证实时检测的前提下,MFANet能够有效地抑制背景噪声对检测器的影响,提升复杂场景下的目标检测精度。

     

    Abstract: Aimed at the problem that the traditional attention model is not effective in complex object detection scenarios, based on the CenterNet algorithm, an anchor-free object detection algorithm based on multi-scale feature aggregation attention is proposed. Multi-scale feature aggregation attention network (MFANet) was used in this algorithm. This attention network not only combined the features of different receptive fields between different blocks of the backbone network, but also paid attention to the important feature regions on the feature map. In addition to this, the keypoint prediction branch of CenterNet was redesigned to use a deeper convolutional network for better predictions. The experimental results on the UAVDT dataset and the COCO dataset show that, on the premise of ensuring real-time detection, the MFANet can effectively suppress the influence of background noise on the detector and improve the object detection accuracy in complex scenes.

     

/

返回文章
返回