摘要:针对X射线安检图像中背景复杂、尺度多变、小尺寸目标难以检测等挑战,提出一种聚焦边缘与多尺度特征的轻量化违禁品检测算法LEM-YOLO。首先,设计轻量化边缘特征增强模块LEFE以构建EFE_C2f,增强模型边缘特征提取能力。其次,设计高效多级特征融合金字塔网络EM-FPN,利用动态上采样Dysample和层次尺度特征金字塔网络HS-FPN,增强多尺度特征融合并减少计算冗余,同时采用动态特征编码模块DFE,保留小尺寸目标的全局信息。最后,使用Shape-IoU作为边界框回归损失函数,聚焦边框形状和自身尺度,提升目标定位精度。实验结果表明,在公开数据集SIXray上,LEM-YOLO的mAP达到了94.63%,比原算法提升了2.56%,同时模型体积下降了50.67%,与同类算法相比能更好满足违禁品检测场景的需求。