arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.06002cs.CV

通过更强引导增强语义分割的生成式数据增强

Enhanced Generative Data Augmentation for Semantic Segmentation via Stronger Guidance

University of Information Technology(信息技术大学);Vietnam National University(越南国家大学)

Quang-Huy Che, Duc-Tri Le, Bich-Nga Pham, Duc-Khai Lam, Vinh-Tiep Nguyen

AI总结:

针对语义分割传统数据增强语义多样性不足的问题,提出基于可控扩散模型的增强流水线,通过类别提示追加等机制提升生成准确性,结合类别平衡算法,在PASCAL VOC数据集上验证了生成高质量合成图像的有效性。

摘要:

数据增强对于语义分割这类像素级标注任务至关重要,此类任务的标注工作需要耗费大量精力与人力。传统方法采用旋转、翻转等简单变换生成新图像,但往往在关键语义维度上缺乏多样性,且无法改变高层语义属性。为解决这一问题,生成式模型已成为通过生成合成图像来增强数据的有效方案。可控生成模型利用原始图像的提示词和视觉参考,为语义分割任务提供数据增强方法。然而,由于难以构建有效的提示词和视觉参考,这类模型在生成能准确反映原始图像内容与结构的合成图像时面临挑战。在本研究中,我们提出了一种基于可控扩散模型的高效语义分割数据增强流水线。我们提出的方法包含通过“类别-提示词追加”和“视觉先验融合”实现的高效提示词生成机制,以增强对真实图像中已标注类别的注意力,使该流水线能够生成精确数量的增强图像,同时保留分割标注类别的结构。此外,我们还实现了一种类别平衡算法,确保合成图像与原始图像合并后训练数据集的类别平衡。在PASCAL VOC数据集上的评估表明,我们的流水线能够有效生成高质量的语义分割合成图像。我们的代码可在https://github.com/chequanghuy/Enhanced-Generative-Data-Augmentation-for-Semantic-Segmentation-via-Stronger-Guidance获取。

英文摘要:

Data augmentation is crucial for pixel-wise annotation tasks like semantic segmentation, where labeling requires significant effort and intensive labor. Traditional methods, involving simple transformations such as rotations and flips, create new images but often lack diversity along key semantic dimensions and fail to alter high-level semantic properties. To address this issue, generative models have emerged as an effective solution for augmenting data by generating synthetic images. Controllable Generative models offer data augmentation methods for semantic segmentation tasks by using prompts and visual references from the original image. However, these models face challenges in generating synthetic images that accurately reflect the content and structure of the original image due to difficulties in creating effective prompts and visual references. In this work, we introduce an effective data augmentation pipeline for semantic segmentation using Controllable Diffusion model. Our proposed method includes efficient prompt generation using Class-Prompt Appending and Visual Prior Blending to enhance attention to labeled classes in real images, allowing the pipeline to generate a precise number of augmented images while preserving the structure of segmentation-labeled classes. In addition, we implement a class balancing algorithm to ensure a balanced training dataset when merging the synthetic and original images. Evaluation on PASCAL VOC datasets, our pipeline demonstrates its effectiveness in generating high-quality synthetic images for semantic segmentation. Our code is available at https://github.com/chequanghuy/Enhanced-Generative-Data-Augmentation-for-Semantic-Segmentation-via-Stronger-Guidance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑