Bounding Box-Guided Diffusion for Synthesizing Industrial Images and Segmentation Map

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Caruso, Emanuele, Simoni, Alessandro, Pelosin, Francesco
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914282108616704
author Caruso, Emanuele
Simoni, Alessandro
Pelosin, Francesco
author_facet Caruso, Emanuele
Simoni, Alessandro
Pelosin, Francesco
contents Synthetic dataset generation in Computer Vision, particularly for industrial applications, is still underexplored. Industrial defect segmentation, for instance, requires highly accurate labels, yet acquiring such data is costly and time-consuming. To address this challenge, we propose a novel diffusion-based pipeline for generating high-fidelity industrial datasets with minimal supervision. Our approach conditions the diffusion model on enriched bounding box representations to produce precise segmentation masks, ensuring realistic and accurately localized defect synthesis. Compared to existing layout-conditioned generative methods, our approach improves defect consistency and spatial accuracy. We introduce two quantitative metrics to evaluate the effectiveness of our method and assess its impact on a downstream segmentation task trained on real and synthetic data. Our results demonstrate that diffusion-based synthesis can bridge the gap between artificial and real-world industrial data, fostering more reliable and cost-efficient segmentation models. The code is publicly available at https://github.com/covisionlab/diffusion_labeling.
format Preprint
id arxiv_https___arxiv_org_abs_2505_03623
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Bounding Box-Guided Diffusion for Synthesizing Industrial Images and Segmentation Map
Caruso, Emanuele
Simoni, Alessandro
Pelosin, Francesco
Computer Vision and Pattern Recognition
Synthetic dataset generation in Computer Vision, particularly for industrial applications, is still underexplored. Industrial defect segmentation, for instance, requires highly accurate labels, yet acquiring such data is costly and time-consuming. To address this challenge, we propose a novel diffusion-based pipeline for generating high-fidelity industrial datasets with minimal supervision. Our approach conditions the diffusion model on enriched bounding box representations to produce precise segmentation masks, ensuring realistic and accurately localized defect synthesis. Compared to existing layout-conditioned generative methods, our approach improves defect consistency and spatial accuracy. We introduce two quantitative metrics to evaluate the effectiveness of our method and assess its impact on a downstream segmentation task trained on real and synthetic data. Our results demonstrate that diffusion-based synthesis can bridge the gap between artificial and real-world industrial data, fostering more reliable and cost-efficient segmentation models. The code is publicly available at https://github.com/covisionlab/diffusion_labeling.
title Bounding Box-Guided Diffusion for Synthesizing Industrial Images and Segmentation Map
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.03623