X-ray Security Contraband Detection on Improved YOLOv7-Tiny

Ya-lin YE , Lian-jun XIE , Bing-peng GAO , Li-jun LÜ

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (26) : 11349 -11357.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (26) : 11349-11357 DOI: 10.12404/j.issn.1671-1815.2400403
research-article
X-ray Security Contraband Detection on Improved YOLOv7-Tiny
Author information +
History +
PDF

Abstract

In response to the issues of low accuracy and poor real-time performance in the X-ray security inspection scenario, a new lightweight detection method has been proposed based on the YOLOv7-Tiny model. Firstly, a lightweight GhostNetV2 module was introduced into the backbone network to reduce model parameters while improving training efficiency. Secondly, a Pyramid Split Attention mechanism was incorporated into the neck network of YOLOv7-Tiny to effectively address the problem of insufficient feature extraction due to parameter reduction, enhancing accuracy in detecting complex backgrounds and multi-scale objects. Finally, the Normalized Wasserstein Distance method was used to measure the loss, replacing the original Intersection over Union metric, which reduces sensitivity to small object position deviations and enhances accuracy in regressing small objects. Experimental results show that the improved model achieves an average detection accuracy of 92. 9%, 76.2%, and 91.2% on the SIXray, CLCXray, and OPIXray datasets, representing 6. 5% , 2% , and 1. 8% improvement over the original algorithm, respectively. The proposed model, while being lightweight, further enhances detection capabilities and can meet real-time detection requirements, demonstrating significant practical value.

Keywords

deep learning / contraband testing / feature extraction / model lightweigh

Cite this article

Download citation ▾
Ya-lin YE, Lian-jun XIE, Bing-peng GAO, Li-jun LÜ. X-ray Security Contraband Detection on Improved YOLOv7-Tiny. Science Technology and Engineering, 2024, 24(26): 11349-11357 DOI:10.12404/j.issn.1671-1815.2400403

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (9463KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉