计算机工程与应用2024，Vol.60Issue(2)：96-102,7.DOI:10.3778/j.issn.1002-8331.2210-0240
自适应与多尺度特征融合的X光违禁品检测
Detection of X-Ray Contraband by Adaptive and Multi-Scale Feature Fusion
摘要
Abstract
To resolve the problems of spatial multi-scale variation,background interference and model complex of X-ray security inspection contraband images,a lightweight YOLOv5 model with spatial adaptation and multi-scale feature fusion is proposed.Taking YOLOv5 as the basic framework,the adaptive spatial feature fusion mechanism is introduced to suppress the influence of feature scale differences,and the bidirectional feature weighted fusion is integrated with the bidirectional feature pyramid network;the lightweight channel attention mechanism is used to obtain accurate position infor-mation and enhance the expression of effective features.Meanwhile,GhostConv is used to replace part of Conv to reduce the computational complexity of the network.This model achieves mAP of 94.2%,92.8%and 83.3%on three public data-sets such as OPIXray,SIXray and HiXray,respectively,which is 5.4,0.5 and 1.7 percentage points higher than the base-line model.And the model training time is not significantly increased.It takes into account the accuracy and speed of model detection,which is superior to many current advanced algorithms.关键词
X光图像/违禁品检测/空间特征融合/YOLOv5Key words
X-ray images/contraband detection/spatial feature fusion/YOLOv5分类
信息技术与安全科学引用本文复制引用
孙嘉傲,董乙杉,郭靖圆,李明泽,李帅超,卢树华..自适应与多尺度特征融合的X光违禁品检测[J].计算机工程与应用,2024,60(2):96-102,7.基金项目
中央高校基本科研业务经费重大项目(2021JKF102) （2021JKF102）
公安学科基础理论研究专项项目(2021XKZX08,2021JC03). （2021XKZX08,2021JC03）
