Contextual Information Fusion for Small Object Detection

In view of the difficulty and low accuracy of small object detection for remote sensing images, this paper proposes a small object detection algorithm based on contextual information fusion to solve the problem of real-time detection accuracy of small object. In this paper, we use bottom-up VGG16 network to realize multi-scale feature extraction to deal with the problem of insufficient image feature extraction. To direct at the problem that the feature information of each feature layer is single, the shallow feature layer and the deep feature layer are fused through the feature fusion module, which achieves the purpose that some feature layers have more abundant fusion features in the structure level. Aiming at the problem that the detection objects in remote sensing images are mainly small and medium-sized objects, this paper proposes to use the multivariate information of four different scale feature layers for classification prediction and regression prediction, so as to reduce the complexity of network model. The experimental results show that the proposed small object detection algorithm based on the fusion of four scale deep and shallow contextual information can obtain good accuracy and real-time performance on the NWPU VHR-10 dataset, improve the detection accuracy on the basis of ensuring the real-time detection, and perform well in the small object detection task of remote sensing images.