基于特征增强和通道权重融合的轻量化小目标检测算法

    Lightweight Small Target Detection Algorithm Based on Feature Enhancement and Channel Weight Fusion

    • 摘要: 本文提出了轻量化的基于特征增强和通道权重融合的小目标检测算法FECWF-YOLO (Feature Enhancement and Channel Weight Fusion YOLO) ,旨在解决无人机等场景中小目标检测精度不足以及模型轻量化程度欠缺的问题。本文首先通过引入小目标检测头以增强小目标捕捉能力,优化网络结构提升检测性能并通过减少骨干网络卷积层数等方式实现轻量化;针对小目标检测头导致计算量增加问题,本文使用部分卷积PConv (Partial Convolution) 模块设计了一种轻量化检测头LDH (Lightweight Detection Head),有效降低计算复杂度。在骨干网络嵌入基于PConv构建的轻量化特征增强模块,利用多支路膨胀卷积扩展感受野,强化小目标特征提取;针对特征融合环节,提出一种基于通道权重融合的特征融合方法,结合可变形卷积DCNv2 (Deformable Convolutional Networks v2) 改善多特征融合效果。实验表明,与基础模型 YOLOv8n,FECWF-YOLO参数量和计算量分别减少了70%和9%,平均精度mAP@0.5(mean Average Precision at IoU threshold 0.5) 提升了5.3个百分点,在性能与效率间取得良好平衡,为无人机等场景下的小目标检测提供有效解决方案。

       

      Abstract: This paper proposes a lightweight small object detection algorithm based on feature enhancement and channel weight fusion, termed FECWF-YOLO (Feature Enhancement and Channel Weight Fusion YOLO), aiming to address the challenges of insufficient detection accuracy for small objects and inadequate model lightweighting in scenarios such as unmanned aerial vehicles. The paper first introduces a small object detection head to enhance the model’s capability in capturing small objects, optimizes the network structure to improve detection performance, and reduces the number of convolutional layers in the backbone network to achieve lightweighting. To mitigate the increased computational cost incurred by the small object detection head, a lightweight detection head (LDH) is designed using the Partial Convolution (PConv) module, effectively reducing computational complexity. Furthermore, a lightweight feature enhancement module based on PConv is embedded in the backbone network, utilizing multi-branch dilated convolutions to expand the receptive field and enhance feature extraction for small objects. In the feature fusion stage, a channel weight fusion-based feature fusion method is proposed, incorporating Deformable Convolutional Networks v2 (DCNv2) to improve the effectiveness of multi-feature fusion. Experimental results show that, compared with the baseline model YOLOv8n, FECWF-YOLO reduces the parameter count and computational complexity by 70% and 9%, respectively, while achieving 5.3 percentage points increase in mAP@0.5. These results indicate a favorable balance between performance and efficiency, providing an effective solution for small object detection in scenarios such as unmanned aerial vehicles.

       

    /

    返回文章
    返回