• 改进YOLO11的X光图像违禁品检测与分割算法

    Improved the contraband detection and segmentation in X-Ray images of YOLO11

    • 针对安检X光图像中违禁物品检测精度不足、深层网络信息易丢失以及部署设备资源受限等挑战,提出一种改进的YOLO11网络模型,用于违禁物品的精确检测与分割。首先,引入DWR模块改进C3k2结构,以扩展感受野并增强高层特征的提取能力。其次,提出Dy-SSFF模块,将DWR模块提取的深层语义信息与违禁品细节特征深度融合,生成多尺度融合特征图,从而抑制关键细节信息的流失,提升模型对多尺度目标的感知能力。最后,采用轻量化共享卷积归一化策略改进分割头,在增强对多尺度特征敏感性的同时,降低模型参数量和计算复杂度。实验结果表明,在Xray-battery数据集上,与原始YOLO11n-seg模型相比,本文模型在预测框上的mAP@0.5和mAP@0.5:0.95分别提升1.5%和3.5%,在掩膜上的mAP@0.5和mAP@0.5:0.95分别提升2.1%和2.8%;同时参数量减少7%,计算量减少8%。所提模型在满足高精度检测与分割要求的同时,可部署于资源受限的设备,为实际安检X光图像场景提供了有效的技术支撑。

       

      Abstract: To address the challenges of insufficient detection accuracy for prohibited items in security inspection X-ray images, the susceptibility to deep network information loss, and the constraints of resource-limited deployment devices, this paper proposes an improved YOLO11 network model for accurate detection and segmentation of prohibited items. First, a DWR (Dilated Wide Residual) module is introduced to enhance the C3k2 structure, expanding the receptive field and strengthening the extraction capability of high-level features. Second, a Dy-SSFF (Dynamic Skip-Spatial Feature Fusion) module is proposed to deeply fuse the high-level semantic information extracted by the DWR module with fine-grained detail features of prohibited items, generating multi-scale fused feature maps. This suppresses the loss of critical detail information and improves the model's perception of multi-scale objects. Finally, a lightweight shared convolutional normalization strategy is adopted to improve the segmentation head, enhancing sensitivity to multi-scale features while reducing the model's parameter count and computational complexity. Experimental results on the Xray-battery dataset demonstrate that, compared to the original YOLO11n-seg model, the proposed model improves mAP@0.5 and mAP@0.5:0.95 by 1.5% and 3.5% for bounding boxes, and by 2.1% and 2.8% for masks, respectively. Simultaneously, the parameter count is reduced by 7%, and the computational cost is reduced by 8%. The proposed model achieves high-precision detection and segmentation while being deployable on resource-constrained devices, providing effective technical support for real-world security inspection X-ray image scenarios.

       

    /

    返回文章
    返回