Intra-class data augmentation with deep generative models of threat objects in baggage radiographs

Abstract

Despite state-of-the-art deep learning-based computer vision models achieving high accuracy on object recognition tasks, x-ray screening of baggage at checkpoints is largely performed by hand. Part of the challenge in automation of this task is the relatively small amount of available labeled training data. Furthermore, realistic threat objects may have forms or orientations that do not appear in any training data, and radiographs suffer from high amounts of occlusion. Using deep generative models, we explore data augmentation techniques to expand the intra-class variation of threat objects synthetically injected into baggage radiographs using openly available baggage x-ray datasets. We also benchmark the performance of object detection algorithms on raw and augmented data.