高级检索

面向遮挡场景的AIFI-SEAM交通标识检测模型

AIFI-SEAM Traffic Sign Detection Model for Occlusion Scenarios

  • 摘要: 精准定位与实时检测交通标识牌是交通设施智慧巡检的核心环节,然而真实道路场景中,远距离交通标识牌易被车辆、行人、树木等物体部分遮挡,严重制约了现有目标检测算法的性能。为解决上述难题,提出一种基于 YOLOv8s 框架的具备遮挡感知的特征交互式鲁棒交通标识检测模型,通过两项关键架构创新提升检测能力。首先,借鉴实时检测变换器(Real-Time Detection Transformer,RT-DETR)的注意力机制,设计注意力引导的尺度内特征交互(Attention-based Intra-scale Feature Interaction,AIFI)模块,将其嵌入 YOLOv8s 颈部网络,通过自适应学习特征图内部上下文信息,生成更具判别性的多尺度语义特征,强化小目标特征表征。其次,针对遮挡问题,引入 YOLO-Face V2 中的分离增强注意力模块(Separated and Enhancement Attention Module,SEAM),构建遮挡感知的 SEAM-Head 检测头,自适应放大被遮挡区域的局部特征权重,提升模型对部分遮挡目标的鲁棒性。在由现实道路采集图像和网上开源图像共同构成,且含部分遮挡样本的交通标识牌数据集上进行训练与评估,实验结果(含消融实验)表明,所提模型各项性能指标均优于基准 YOLOv8s,其中 mAP@0.5:0.95 指标提升 11.03个百分点,在小尺寸和部分遮挡交通标识牌检测任务中表现尤为突出,充分验证了 AIFI 模块与 SEAM-Head 设计的有效性,为复杂道路场景下的交通标识牌智慧巡检提供了可靠技术支撑。

     

    Abstract: Accurate positioning and real-time detection of traffic signs are the core links of intelligent inspection of transportation facilities. However, in real road scenarios, long-distance traffic signs are easily partially occluded by vehicles, pedestrians, trees and other objects, which seriously restricts the performance of existing object detection algorithms. To solve the above problem, an occlusion-aware feature-interactive robust traffic sign detection model based on the YOLOv8s framework is proposed, with detection capability improved through two key architectural innovations. Firstly, drawing on the attention mechanism of Real-Time Detection Transformer (RT-DETR), an Attention-Guided Intra-Scale Feature Interaction (AIFI) module is designed and embedded into the neck network of YOLOv8s. By adaptively learning the internal contextual information of feature maps, it generates more discriminative multi-scale semantic features and enhances the representation of small target features. Secondly, to address the occlusion issue, the Separated Enhanced Attention Module (SEAM) from YOLO-Face V2 is introduced to construct an occlusion-aware SEAM-Head detection head. It adaptively amplifies the local feature weights of occluded regions, improving the model’s robustness to partially occluded targets. Training and evaluation are conducted on a traffic sign dataset composed of real-world road images and online open-source images and containing partially occluded samples. Experimental results (including ablation experiments) show that all performance metrics of the proposed model are superior to the baseline YOLOv8s, with the mAP@0.5:0.95 metric improved by 11.03 percentage points. It performs particularly well in the detection tasks of small-sized and partially occluded traffic signs, fully verifying the effectiveness of the AIFI module and SEAM-Head design. This provides reliable technical support for the intelligent inspection of traffic signs in complex road scenarios.

     

/

返回文章
返回