Test2train Amine Face

🚀 初めての事前学習モデル -- 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

📄 ライセンス

このモデルはMITライセンスの下で提供されています。