Research Article

A Data Augmentation Method for Prohibited Item X-Ray Pseudocolor Images in X-Ray Security Inspection Based on Wasserstein Generative Adversarial Network and Spatial-and-Channel Attention Block

Table 2

The detection results.

ItemsYOLOV4-tiny + the original training datasetYOLOV4-tiny + the augmented training dataset

Brass knuckle88.75%95.76%
Firecracker86.63%92.88%
Gun86.80%93.84%
Hammer67.81%80.59%
Knife74.20%80.58%
Lighter86.69%94.81%
Mental bottle93.17%96.23%
Plier89.77%90.11%
Scissors72.79%87.06%
mAP82.96%90.21%