Improved YOLOv5 Model for X-Ray Prohibited Item Detection

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

Download full size

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

Download full size

Table 1. Comparison indicators of the two models before and after improvement
View table
Table 1. Comparison indicators of the two models before and after improvement
Method mAP /% G /% K /% W /% P /% S /% Size /MB
YOLOv5s 87.2 98.1 79.2 84.8 90.3 83.8 14.04
Ours（5s） 89.6 98.4 82.2 88.4 92.2 86.7 15.72
YOLOv5m 89.5 98.2 83.1 87.6 94.2 84.5 41.46
Ours（5m） 92.5 99.1 85.6 90.5 95.2 92.0 45.18

Table 2. Comparison results on the SIXray dataset
View table
Table 2. Comparison results on the SIXray dataset
Method mAP /%
YOLOv3^［29］ 79.2
YOLOv4^［30］ 83.1
ASPP-YOLOv4^［31］ 85.2
SSD^［32］ 82.9
YOLOv5s^［26］ 87.2
YOLOv5s+Ours 89.6

Table 3. Comparison results on the OPIXray and HiXray datasets

View table

Table 3. Comparison results on the OPIXray and HiXray datasets

Method	OPIXray mAP /%	HiXray mAP /%
YOLOv3^［29］	78.2	-
YOLOv3+DOAM^［20］	79.2	-
YOLOv4^［30］	78.9	-
CHR^［18］	78.6	-
FBS^［19］	81.7	-
SSD^［32］	70.9	71.4
SSD+DOAM^［20］	74.0	72.1
SSD+LIM^［28］	74.6	73.1
FCOS^［33］	82.0	75.7
FCOS+DOAM^［20］	82.4	76.2
FCOS+LIM^［28］	83.1	77.3
YOLOv5s^［26］	87.8	81.7
YOLOv5s+DOAM^［20］	88.0	82.2
YOLOv5s+LIM^［28］	90.6	83.2
YOLOv5s+Ours	91.6	83.1

Table 4. Ablation study on the SIXray dataset

View table

Table 4. Ablation study on the SIXray dataset

Component	YOLOv5m		YOLOv5s
Component	mAP /%	Size /MB	mAP /%	SIZE /MB
Baseline	89.5	41.46	87.2	14.04
Baseline+CBAM	89.7	45.18	88.6	15.72
Baseline+Mixup	90.4	41.46	87.6	14.04
Baseline+Mixup+CBAM	90.5	45.18	88.0	15.72
Baseline+WBF	90.2	41.46	88.2	14.04
Baseline+WBF+CBAM	90.1	45.18	89.1	15.72
Baseline+WBF+Mixup	91.0	41.46	89.4	14.04
Baseline+WBF+Mixup +CBAM	92.5	45.18	89.6	15.72

Tools

Get Citation

Copy Citation Text

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

Download Citation

EndNote(RIS)BibTex Plain Text

Set citation alerts for article

Save article for my favorites