license: mit tags:

• pytorch
• diffusers
• unconditional-image-generation

中文版

我的首个预训练模型——test2train_anime_face模型卡

这是一个用于生成64*64像素动漫风格面部图像的无条件扩散模型。
训练数据集采用anime-faces，该数据集包含从www.getchu.com抓取的21551张动漫面部图像，并使用https://github.com/nagadomi/lbpcascade_animeface中的动漫面部检测算法进行了裁剪处理。

批量生成图像时容易出现面部扭曲现象。经测试，单次生成一张图像效果最佳，可有效避免面部失真问题。

使用方法

from diffusers import DDPMPipeline

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
pipeline = DDPMPipeline.from_pretrained('Chilli-b/test2train_amine_face').to(device)
image = pipeline().images[0]
image