Test2train Amine Face

🚀 我的首個預訓練模型——test2train_anime_face

本模型是一個用於無條件生成 64*64 尺寸動漫風格臉部圖片的擴散模型。它藉助特定的訓練集，能夠生成風格獨特的動漫臉圖片。不過在生成時，一次生成多張圖片容易出現鬼臉，每次生成一張圖片效果最佳。

🚀 快速開始

該模型是一個無條件擴散模型，用於生成尺寸為 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

📄 許可證

本項目採用 MIT 許可證。