Small object detection is a very challenging task in the field of object detection because it is easily affected by large object occlusion and small object itself has relatively little feature information. Aiming at the problem that the YOLOv3 network does not consider the context semantic relationship of small object detection, the detection accuracy of small objects is not high. In this paper, we propose a small object detection network combining multi-level fusion and feature augmentation. First, the feature enhancement module is introduced into the deep layer of the backbone extraction network to enhance the feature information of small objects in the feature map. Second, a multi-level feature fusion module is proposed to better capture the contextual semantic relationship of small objects. In addition, the strategy of combining Soft-NMS and CIOU is used to solve the problem of missed detection of occluded small objects. At last, The ablation experiment of the MS COCO2017 object detection task proves the effectiveness of several modules introduced in this paper for small object detection. The experimental results on the MS COCO2017, VOC2007, and VOC2012 datasets show that the AP of this method is 16.5%, 8.71%, and 9.68% higher than that of YOLOv3, respectively. All experiments show that the method proposed in this paper has better detection performance for small object detection.
翻译:在物体探测领域,小型物体探测是一项非常具有挑战性的任务,因为它很容易受到大型物体隔离的影响,而小型物体本身的特性信息相对较少。针对YOLOv3网络不考虑小物体探测的上下文语义关系的问题,小物体的探测精度并不高。在本文件中,我们提议建立一个小型物体探测网络,将多级聚合和特性增强结合起来。首先,特性增强模块被引入主干提取网络的深层中,以加强特征图示中小物体的特征信息。第二,提出一个多级特性聚合模块,以更好地捕捉小物体的上下文语义关系。此外,将软-NMS和CIOUU相结合的战略用于解决未探测到的小型物体的问题。最后,MS COCO2017 物体探测任务的放大实验证明了本文中引入的几个模块的有效性。MS COCO2017、 VOC2007和VOC2012天物体的实验结果,以更好地捕捉到小物体的背景语系关系。此外,这一方法的性能率为16.5%,用于检测。