微调稳定扩散模型的裂缝图像数据集扩充方法

吴杰,韩贝林,张舣航,邹超,辛莉峰,黄仕平

Method for augmenting crack image datasets via fine-tuning of stable diffusion models

Jie WU,Beilin HAN,Yihang ZHANG,Chao ZOU,Lifeng XIN,Shiping HUANG

表 1 不同模型在裂缝图像生成任务中的定量分析结果

Tab.1 Quantitative analysis of different models for crack image generation

模型 IS(↑) FID(↓) KID(↓) LPIPS(↓) 训练集 2.996 — — — DCGAN 1.232 314.225 0.355 0.646 WGAN-GP 1.510 162.939 0.143 0.420 StyleGAN 2.016 150.253 0.120 0.418 微调前的SD 2.993 292.201 0.009 0.582 本研究 2.989 116.624 0.055 0.390