嵌入自适应空间注意力的Scaled-YOLOv4小目标检测模型

SCALED-YOLOv4 MODEL EMBEDDED WITH ADAPTIVE SPATIAL ATTENTION FOR SMALL OBJECT DETECTION

  • 摘要: 针对目标检测方法中网络采用固定感受度传感模拟特征时只关注常规尺寸目标而忽略小目标的特征造成检测精度低的问题,提出自适应空间注意力机制,增加并行的不同大小卷积核,嵌入Scaled-YOLOv4残差结构的3×3卷积层中,使网络根据不同的尺寸的物体自主调节感受度大小加强对小目标特征的提取。实验结果表明,新的网络模型能有效提升小目标的检测精度,并改善原模型存在的误检和漏检问题。在MSCOCO和PASCAL VOC等数据集上的检测精度均比之前有较大提升。

     

    Abstract: In order to solve the problem of low detection accuracy caused by fixed receptive field in object detection while convolution only pays attention to conventional size targets and ignores the characteristics of small targets, an adaptive spatial attention mechanism is proposed. This method added parallel convolution kernels of different sizes and was embedded in the 3×3 convolution layer of Scaled-YOLOv4 residual structure, so that the network could adjust the receptive field size according to different sizes to enhance the feature extraction of small targets. The experimental results show that the new network model can effectively improve the detection accuracy of the algorithm for small targets, and improve the problems of false detection and missed detection in the original model. The detection accuracy on datasets such as MSCOCO and PASCAL VOC has been greatly improved.

     

/

返回文章
返回