基于改进YOLOv3的小目标检测算法

Small object detection algorithm based on improved YOLOv3

  • 摘要: 为有效解决小目标难以召回、易发生漏检的问题,提出一种基于特征融合和特征增强的YOLOv3改进算法。为增强模型的泛化性能,训练时利用Mosaic和Mixup方法联合数据增强。首先,为改善小目标检测召回率低的问题,延伸原特征融合网络至更浅层,并添加自底向上特征金字塔,使浅层特征层的细节和定位信息更多地传递至深层;其次,提出一种特征增强模块,增大感受野,使浅层特征层获得丰富的深层语义信息,优化特征层的表达能力;最后,将GIoU作为回归损失函数,以降低漏检率,实现更准确的回归。在Pascal VOC2007和VOC2012上进行仿真实验,实验结果表明,改进算法在保证检测速度的前提下,mAP(mean average precision)值提高4.4%。结果充分证明本文算法能有效提升小目标检测性能。

     

    Abstract: In order to effectively solve the problem that small objects are difficult to recall and easy to miss detection, an improved YOLOv3 algorithm based on feature fusion and feature enhancement was proposed. To enhance the generalization performance of the model, the Mosaic and Mixup methods were combined for data enhancement during training. Firstly, in order to improve the low recall rate of small object detection, the original feature fusion network was extended to the shallower layer, and the bottom-up feature pyramid was added, so that the details and positioning information of the shallow feature layer could be transmitted to the deep layer. Secondly, a feature enhancement module was proposed to enlarge the receptive field, so that the shallow feature layer could obtain the rich deep semantic information and optimize the expression ability of the feature layer. Finally, the GIoU was used as a regression loss function to reduce the missing rate and achieve more accurate regression. Simulation experiments on Pascal VOC2007 and VOC2012 show that the improved algorithm can improve the mean average precision (mAP) value by 4.4% on the premise of ensuring the detection speed. Experimental results fully prove that the proposed algorithm can effectively improve the performance of small object detection.

     

/

返回文章
返回