基于WHIoU损失的安检图像违禁物品检测算法

朱雷, 李国权, 武瑞恒, 黄正文, 庞宇

重庆邮电大学学报(自然科学版) ›› 2025, Vol. 37 ›› Issue (01) : 46-56.

基于WHIoU损失的安检图像违禁物品检测算法

  • 朱雷, 李国权, 武瑞恒, 黄正文, 庞宇
作者信息 +
History +

摘要

针对X光安检图像存在的违禁物品重叠和背景复杂等问题,提出了一种基于YOLOv7改进的X光安检图像违禁物品检测算法。在模型颈部网络中构建注意力特征融合模块,分别添加空间注意力和通道注意力挖掘浅层细节特征和深层语义特征,减少噪音信息冗余,降低有效特征丢失;重新设计WHIoU Loss替换CIoU Loss作为新的边界框损失函数,确保在预测框与真实边界框的宽高呈现相同线性比例关系的情况下,宽高比惩罚函数仍然具有约束效果,提升收敛速度和精度。在SIXray数据集上的实验结果表明,算法在mAP50、mAP50:95上分别提高了2%和4.5%,同时检测速度也达到了58 FPS。

关键词

深度学习 / 目标检测 / YOLOv7 / 安检图像 / 注意力机制

中图分类号

TP18 / TP391.41 / X924.2

引用本文

导出引用
朱雷, 李国权, 武瑞恒, 黄正文, 庞宇. 基于WHIoU损失的安检图像违禁物品检测算法. 重庆邮电大学学报(自然科学版). 2025, 37(01): 46-56

基金

国家自然科学基金项目(U21A20447); 重庆市创新群体科学基金项目(cstc2020jcyj-cxttX0002)

评论

Accesses

Citation

Detail

段落导航
相关文章

/