The task of steel surface defect recognition is an industrial problem with
great industry values. The data insufficiency is the major challenge in
training a robust defect recognition network. Existing methods have
investigated to enlarge the dataset by generating samples with generative
models. However, their generation quality is still limited by the insufficiency
of defect image samples. To this end, we propose Stable Surface Defect
Generation (StableSDG), which transfers the vast generation distribution
embedded in Stable Diffusion model for steel surface defect image generation.
To tackle with the distinctive distribution gap between steel surface images
and generated images of the diffusion model, we propose two processes. First,
we align the distribution by adapting parameters of the diffusion model,
adopted both in the token embedding space and network parameter space. Besides,
in the generation process, we propose image-oriented generation rather than
from pure Gaussian noises. We conduct extensive experiments on steel surface
defect dataset, demonstrating state-of-the-art performance on generating
high-quality samples and training recognition models, and both designed
processes are significant for the performance.

针对钢表面缺陷图像生成的数据不足问题，提出了一种稳定的钢表面缺陷图像生成方法，通过在生成过程中对数据分布进行参数调整和采用钢表面图像导向的生成方法，以解决钢表面图像与生成图像的分布差异问题，并在钢表面缺陷数据集上展示了出色的样本生成和识别模型训练性能。