融合边界上下文协同与门控回归的小目标检测算法

Small Object Detection Algorithm Integrating Boundary-Context Collaboration and Gated Regression

  • 摘要: 针对无人机航拍图像中小目标占比高、边界弱以及中心偏移导致定位误差等问题,提出一种面向无人机航拍图像的小目标检测算法FHG-YOLO。首先,在主干网络中引入特征互补映射(Feature Complementary Mapping,FCM)和渐进式多核深度卷积(Progressive Multi-kernel Depthwise Convolution,PzConv),减少小目标特征丢失;其次,设计动态选择空间金字塔池化(Dynamic Selective Spatial Pyramid Pooling-Fast,DSSPF),通过动态尺度选择和轻量上下文建模,增强上下文表达能力。然后,提出头部边界-上下文混合(Head Boundary-Context Mixer,HBCM)模块,通过细节、上下文和边界分支构建双向协同机制,并采用竞争式融合,增强复杂背景下小目标边界判别能力;最后,提出门控中心归一化损失函数(WIoU-v2 guided NWD with Adaptive Weighting Loss,WNW-Loss),将WIoUv2与Center-NWD结合,并引入尺度门控和质量门控,从而提高小目标定位稳定性。实验结果表明,在VisDrone2019数据集上,FHG-YOLO的精确率和mAP50分别提升3.47和2.55个百分点,参数量降低65.8%;在DOTA数据集上的实验验证了算法具有一定泛化能力。。

     

    Abstract: UAV aerial images suffer from blurred object boundaries, complex background interference, dense occlusion, and high localization sensitivity caused by center offsets. To address these problems, this paper proposes a lightweight UAV small object detection algorithm based on YOLOv8n, named FHG-YOLO. First, Feature Complementary Mapping (FCM) and Progressive Multi-kernel Depthwise Convolution (PzConv) are introduced into the backbone network to enhance the complementary interaction between spatial details and semantic features, thereby reducing the loss of small-object information during downsampling. Second, a Dynamic Selective Spatial Pyramid Pooling-Fast (DSSPF) module is designed to adaptively select effective contextual scales and enhance long-range contextual representation. The main innovations of this paper lie in the proposed Head Boundary-Context Mixer (HBCM) and WNW-Loss. Unlike conventional multi-branch feature fusion, HBCM establishes a collaborative mechanism among detail, context, and boundary information. Specifically, boundary responses are used to actively guide detail enhancement, encouraging the network to focus on discriminative texture regions around object contours. Meanwhile, contextual semantics are introduced to refine boundary responses and suppress misleading background edges. On this basis, a competitive branch fusion strategy is designed to adaptively allocate the contributions of detail, context, and boundary features according to the input feature distribution, enabling the model to dynamically select more effective information under different object scales, occlusion degrees, and background complexities. In addition, a WIoU-v2 guided NWD with Adaptive Weighting Loss, termed WNW-Loss, is proposed for bounding box regression. This loss function combines WIoU-v2 with Center-NWD and introduces scale and quality gates to strengthen the center constraint mainly for small-scale objects and low-quality samples, while avoiding unnecessary interference with high-quality regression samples. Experimental results on the VisDrone2019 dataset show that FHG-YOLO achieves 47.33% precision and 36.19% mAP50, which are 3.47 and 2.55 percentage points higher than those of YOLOv8n, respectively, while reducing the number of parameters by 65.8%. Experiments on the DOTA dataset further verify the generalization capability of the proposed method. The results demonstrate that FHG-YOLO improves small object detection accuracy and localization stability while maintaining a lightweight structure, making it suitable for resource-constrained UAV aerial image detection tasks.

     

/

返回文章
返回