• 融合渐进式尺度建模与可变形注意力的小目标交通标志检测算法

    Small traffic sign detection algorithm integrating progressive scale-aware spatial modeling and deformable attention

    • 针对小目标检测中特征提取与定位困难的问题,提出了一种面向小目标检测任务的YOLO11改进算法PSSM-YOLO。该方法以 YOLO11n 为基线,引入YOLOv8的C2f结构,并融合渐进式尺度建模模块(Progressive Scale-aware Spatial Modeling, PSSM),通过多分支特征融合与多感受野逐通道卷积,有效保留细粒度信息并结合上下文信息,增强模型对小目标的特征表征能力。同时,设计了融合可变形注意力机制的C2PSA-DA模块,引入动态采样策略以自适应聚焦关键目标区域并抑制冗余背景,增强了网络在复杂交通场景中对小目标及其上下文信息的特征提取能力。在Neck中新增1/4下采样的P2检测层,使浅层细粒度特征直接参与预测,减少多次下采样带来的小目标信息损失,进一步提升检测性能。引入WIoU v3损失函数,利用动态非单调聚焦机制强化中等质量样本的学习,抑制低质量样本干扰。实验结果表明:在CCTSDB2021数据集上,PSSM-YOLO在mAP@50和mAP@50:95指标上分别较YOLO11n提升了4.5%和3.6%,整体检测精度与鲁棒性显著提高。

       

      Abstract: To address the challenges of feature extraction and localization in small object detection, this paper proposes an improved YOLO11 algorithm, termed PSSM-YOLO, specifically designed for small object detection tasks. The method adopts YOLO11n as the baseline, incorporates the C2f structure from YOLOv8, and integrates a Progressive Scale-aware Spatial Modeling (PSSM) module. Through multi-branch feature fusion and multi-receptive-field channel-wise convolution, the approach effectively preserves fine-grained information while incorporating contextual cues, thereby enhancing the model's feature representation capability for small objects. Furthermore, a C2PSA-DA module integrating deformable attention mechanism is designed, which introduces a dynamic sampling strategy to adaptively focus on critical target regions and suppress redundant background, thus strengthening the network's feature extraction capability for small objects and their contextual information in complex traffic scenes. A P2 detection layer with 1/4 downsampling is newly added to the Neck, enabling shallow fine-grained features to directly participate in prediction and reducing the loss of small object information caused by multiple downsampling operations, thereby further improving detection performance. In addition, the WIoU v3 loss function is introduced, which employs a dynamic non-monotonic focusing mechanism to reinforce the learning of medium-quality samples while suppressing the interference from low-quality samples. Experimental results on the CCTSDB2021 dataset demonstrate that PSSM-YOLO achieves significant improvements in overall detection accuracy and robustness, outperforming the baseline YOLO11n by 4.5% in mAP@50 and 3.6% in mAP@50:95.

       

    /

    返回文章
    返回