Nguyen Hong Duc, Cai Rizhao, Zhao Heng, Kot Alex C, Wen Bihan
School of Electrical & Electronic Engineering, Nanyang Technological University, Singapore 639798, Singapore.
Micromachines (Basel). 2022 Mar 31;13(4):565. doi: 10.3390/mi13040565.
X-ray imaging machines are widely used in border control checkpoints or public transportation, for luggage scanning and inspection. Recent advances in deep learning enabled automatic object detection of X-ray imaging results to largely reduce labor costs. Compared to tasks on natural images, object detection for X-ray inspection are typically more challenging, due to the varied sizes and aspect ratios of X-ray images, random locations of the small target objects within the redundant background region, etc. In practice, we show that directly applying off-the-shelf deep learning-based detection algorithms for X-ray imagery can be highly time-consuming and ineffective. To this end, we propose a Task-Driven Cropping scheme, dubbed TDC, for improving the deep image detection algorithms towards efficient and effective luggage inspection via X-ray images. Instead of processing the whole X-ray images for object detection, we propose a two-stage strategy, which first adaptively crops X-ray images and only preserves the task-related regions, i.e., the luggage regions for security inspection. A task-specific deep feature extractor is used to rapidly identify the importance of each X-ray image pixel. Only the regions that are useful and related to the detection tasks are kept and passed to the follow-up deep detector. The varied-scale X-ray images are thus reduced to the same size and aspect ratio, which enables a more efficient deep detection pipeline. Besides, to benchmark the effectiveness of X-ray image detection algorithms, we propose a novel dataset for X-ray image detection, dubbed SIXray-D, based on the popular SIXray dataset. In SIXray-D, we provide the complete and more accurate annotations of both object classes and bounding boxes, which enables model training for supervised X-ray detection methods. Our results show that our proposed TDC algorithm can effectively boost popular detection algorithms, by achieving better detection mAPs or reducing the run time.
X射线成像机器广泛应用于边境管制检查站或公共交通中,用于行李扫描和检查。深度学习的最新进展使得能够对X射线成像结果进行自动目标检测,从而大幅降低人工成本。与自然图像上的任务相比,X射线检查的目标检测通常更具挑战性,这是由于X射线图像的大小和宽高比各不相同、小目标物体在冗余背景区域内的位置随机等原因。在实践中,我们发现直接将现成的基于深度学习的检测算法应用于X射线图像可能会非常耗时且效率低下。为此,我们提出了一种任务驱动裁剪方案,称为TDC,用于改进深度图像检测算法,以通过X射线图像实现高效且有效的行李检查。我们不是处理整个X射线图像进行目标检测,而是提出了一种两阶段策略,该策略首先自适应裁剪X射线图像,只保留与任务相关的区域,即用于安全检查的行李区域。使用特定任务的深度特征提取器来快速识别每个X射线图像像素的重要性。只保留有用的且与检测任务相关的区域,并将其传递给后续的深度检测器。这样,不同尺度的X射线图像就被缩小到相同的大小和宽高比,从而实现更高效的深度检测流程。此外,为了评估X射线图像检测算法的有效性,我们基于流行的SIXray数据集提出了一个用于X射线图像检测的新颖数据集,称为SIXray-D。在SIXray-D中,我们提供了完整且更准确的物体类别和边界框注释,这使得能够对监督式X射线检测方法进行模型训练。我们的结果表明,我们提出的TDC算法能够通过获得更好的检测平均精度均值(mAPs)或减少运行时间,有效地提升流行的检测算法。