卡通肖像扩散模型

Stable Diffusion v1.5 在由 3D-CariGAN 剪裁为512x512并添加了标注的 2D 卡通肖像数据集上进行了微调

如果您想了解如何生成自己的带有标注的数据集详细信息，请参阅此 colab

使用 Hugging-Face 的文本到图像训练工具完成了训练 script

关于

输入文本提示并生成卡通肖像

AUTO1111 webui 检查点

main 文件夹中包含一个 .safetensors 文件，将其放入 stable-diffusion-webui "stable-diffusion-webui/models/Stable-diffusion" 文件夹中，并用于生成图像

数据集和训练

在 runwayml/stable-diffusion-v1-5 上进行了10,000次迭代的微调，使用家用台式电脑上的1xA5000 GPU进行了训练

由 @Norod78 训练

原始卡通肖像数据集引用资料

@article{ye2021caricature,
 author = {Ye, Zipeng and Xia, Mengfei and Sun, Yanan and Yi, Ran and Yu, Minjing and Zhang, Juyong and Lai, Yu-Kun and Liu, Yong-Jin},
 title = {3D-CariGAN: An End-to-End Solution to 3D Caricature Generation from Normal Face Photos},
 journal = {IEEE Transactions on Visualization and Computer Graphics},
 year = {2021},
 doi={10.1109/TVCG.2021.3126659},
}