AIGDet: Altitude-Information Guided Vehicle Target Detection in UAV-Based Images

Conducting extensive vehicle detection through the high-altitude perspective offered by Unmanned Aerial Vehicle (UAV) poses significant challenges. The high-altitude operation of UAV to acquire a broader reconnaissance view results in low-resolution and densely packed vehicle targets in the captured imagery, creating substantial difficulties for vehicle detection. To address this, we propose a vehicle detection network specifically designed for UAV, incorporating an end-to-end network that takes scale consistency constraints into consideration. The cornerstone of our method is the Dynamic Feature Refinement Module (DFRM), designed to overcome the feature attenuation and limitations in utilizing high-level priori information common in traditional approaches. Initially, we developed an adaptive target suggestion module based on the priori characteristics of the targets and scenes, and the scale consistency hypothesis of similar vehicles at different UAV flying altitudes. This module optimizes the number and scale of anchors by introducing priori information, facilitating preliminary localization of small-scale imaging targets. Subsequently, we constructed a multi-layer feature purification structure based on a Feature Pyramid Network (FPN) to refine bounding boxes at each level with height priori, integrating additional contextual information. This approach allows us to utilize more contextual information for vehicle detection while enhancing localization accuracy through detailed height priori. Our application and evaluation on multiple open-source datasets with height labels demonstrate that our method, with minimal parameter introduction, achieves excellent mean average precision (mAP) value. This underscores the effectiveness of our approach in UAV-based vehicle detection.