arXiv:2511.15299cs.CV2025-11被引 1

用单阶段生成技术高效合成高质量X光安检图像,无需额外标注劳动。

Taming Generative Synthetic Data for X-ray Prohibited Item Detection

  • 基于文本到图像的单阶段生成,避免传统两阶段中的繁琐前景提取。
  • 合成图像使检测模型平均精度提升1.2%,在多个数据集上有效。
  • 适合需要大量真实感安检图像的工业级安全检测系统研发者。

训练违禁品检测模型需要大量X光安检图像,但收集和标注耗时费力。为解决数据不足问题,现有图像合成方法通过拼接图像扩充数据集,但多采用两阶段流程:先人工提取前景,再进行图像合成,导致额外人力成本且效率低下。本文提出一种基于文本到图像生成的一阶段合成管道(Xsyn),引入两项有效策略提升合成图像质量:交叉注意力精修(CAR)利用扩散模型的交叉注意力图优化边界框标注;背景遮挡建模(BOM)在隐空间显式建模背景遮挡,增强成像复杂性。据我们所知,Xsyn是首个在无需额外人工标注的前提下实现高质量合成的方案。实验表明,该方法在所有对比方法中取得最高性能,带来1.2%的mAP提升,且合成图像能有效提升多种X光安检数据集与检测器上的违禁品检测效果。代码已开源:https://github.com/pILLOW-1/Xsyn/

原文摘要 · Abstract (English)

Training prohibited item detection models requires a large amount of X-ray security images, but collecting and annotating these images is time-consuming and laborious. To address data insufficiency, X-ray security image synthesis methods composite images to scale up datasets. However, previous methods primarily follow a two-stage pipeline, where they implement labor-intensive foreground extraction in the first stage and then composite images in the second stage. Such a pipeline introduces inevitable extra labor cost and is not efficient. In this paper, we propose a one-stage X-ray security image synthesis pipeline (Xsyn) based on text-to-image generation, which incorporates two effective strategies to improve the usability of synthetic images. The Cross-Attention Refinement (CAR) strategy leverages the cross-attention map from the diffusion model to refine the bounding box annotation. The Background Occlusion Modeling (BOM) strategy explicitly models background occlusion in the latent space to enhance imaging complexity. To the best of our knowledge, compared with previous methods, Xsyn is the first to achieve high-quality X-ray security image synthesis without extra labor cost. Experiments demonstrate that our method outperforms all previous methods with 1.2% mAP improvement, and the synthetic images generated by our method are beneficial to improve prohibited item detection performance across various X-ray security datasets and detectors. Code is available at https://github.com/pILLOW-1/Xsyn/.

X光安检图像合成扩散模型数据增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。