arXiv:2501.02264cs.CV2025-01被引 3

用AI自动生成带分割图的新类别图像,低成本扩充语义分割数据集

Unsupervised Class Generation to Expand Semantic Segmentation Datasets

  • 基于Stable Diffusion和SAM的无监督生成流程,自动产出带标签的新类图像
  • 新类平均达到51%交并比,同时提升原有类别的分割准确率
  • 适合需要快速扩展数据集的研究者,尤其适用于少样本场景

语义分割需对图像进行像素级分类,标注成本高。当前常用合成数据结合领域自适应方法来降低标注负担,但这类数据为封闭集合,新增类别需修改生成工具,而该工具往往不公开。近年来,扩散模型如Stable Diffusion已能实现文本到图像的高质量生成。本文提出一种无监督流程,利用Stable Diffusion与Segment Anything Module生成带分割掩码的类别图像,并通过剪切融合方式将新类加入现有数据集,仅需极少人工干预。实验表明,该方法可有效提升无监督域适应性能:新类别平均取得51%的交并比(IoU),且原有类别的错误率下降,整体分割效果更优。

原文摘要 · Abstract (English)

Semantic segmentation is a computer vision task where classification is performed at a pixel level. Due to this, the process of labeling images for semantic segmentation is time-consuming and expensive. To mitigate this cost there has been a surge in the use of synthetically generated data -- usually created using simulators or videogames -- which, in combination with domain adaptation methods, can effectively learn how to segment real data. Still, these datasets have a particular limitation: due to their closed-set nature, it is not possible to include novel classes without modifying the tool used to generate them, which is often not public. Concurrently, generative models have made remarkable progress, particularly with the introduction of diffusion models, enabling the creation of high-quality images from text prompts without additional supervision. In this work, we propose an unsupervised pipeline that leverages Stable Diffusion and Segment Anything Module to generate class examples with an associated segmentation mask, and a method to integrate generated cutouts for novel classes in semantic segmentation datasets, all with minimal user input. Our approach aims to improve the performance of unsupervised domain adaptation methods by introducing novel samples into the training data without modifications to the underlying algorithms. With our methods, we show how models can not only effectively learn how to segment novel classes, with an average performance of 51% IoU, but also reduce errors for other, already existing classes, reaching a higher performance level overall.

语义分割数据增强生成模型无监督

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