VMC-Net: multi-scale feature aggregation and distribution with contextual attention guided fusion for aerial object detection
摘要
As an important branch of remote sensing technology, aerial image target detection plays an indispensable role in supporting urban planning, disaster assessment, and other fields. However, this task faces many challenges such as small object size and complex background, which increase the difficulty of detection. Existing methods usually use multi-scale feature fusion or attention mechanism to improve performance, but they often ignore the role of object feature perception in the image and have problems such as insufficient use of context information. To address these problems, we propose the VMC-Net framework to optimize the aerial image object detection task. The VHeat C2f module enhances the feature extraction capability and generates a clearer target feature map; the multi-scale feature aggregation and distribution module adds feature distribution technology on the basis of the multi-scale feature fusion strategy to achieve more effective scale interaction; the contextual attention guided fusion module uses attention mechanism and weighted fusion method to effectively utilize context information and significantly improve the performance of small object detection. We evaluate the VMC-Net framework on the AI-TOD, VisDrone-2019 and TinyPerson datasets. Experimental results show that our framework outperforms the mainstream target detection methods in the past three years in aerial object detection, with mAP50 scores of 45.6%, 45.9%, and 25.4% respectively.