Yishan Dong, Zhaoxin Li, Jingyuan Guo, Tianyu Chen, Shuhua Lu. Improved YOLOv5 Model for X-Ray Prohibited Item Detection[J]. Laser & Optoelectronics Progress, 2023, 60(4): 0415005

Search by keywords or author
- Laser & Optoelectronics Progress
- Vol. 60, Issue 4, 0415005 (2023)

Fig. 1. Network structure. (a) Improved YOLOv5 model; (b) FPN and PAN

Fig. 2. Structure of CBAM attention module

Fig. 3. Schematic diagrams of WBF and NMS algorithms

Fig. 4. Example of Mixup data augmentation

Fig. 5. Detection results of the two models before and after improvement on the SIXray dataset

Fig. 6. Confusion matrix and P-R curve of proposed method on SIXray, HiXray, OPIXray dataset. FP for background false positive. (a) Confusion matrix on SIXray; (b) confusion matrix on OPIXray; (c) confusion matrix on HiXray; (d) P-R curves on SIXray; (e) P-R curves on OPIXray; (f) P-R curves on HiXray

Fig. 7. Example of bounding boxes which are taken from SIXray dataset. (a) Original images ; (b) bounding boxes generated by NMS; (c) bounding boxes generated by WBF
|
Table 1. Comparison indicators of the two models before and after improvement
|
Table 2. Comparison results on the SIXray dataset
|
Table 3. Comparison results on the OPIXray and HiXray datasets
|
Table 4. Ablation study on the SIXray dataset

Set citation alerts for the article
Please enter your email address