AnimeGANv2に、かわいいトレーニング済みモデルが出てきて、
しかも、Torch-Hubでモデルをダウンロードできるようになって、使いやすくなった。
なにせ、以下のコードだけで写真をアニメ風に変換できる。
from PIL import Image
import torch
model = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v2")
face2paint = torch.hub.load("bryandlee/animegan2-pytorch:main", "face2paint", size=512)
img = Image.open(input_image_path).convert("RGB")
out = face2paint(model, img)
事前トレーニング済みの重みには2021/12現在以下のバリエーションがある。
| face_paint_512_v2 | celeba_distill | face_paint_512_v1 | paprika |
|---|---|---|---|
|
 |
 |
 |
Colabデモ：AnimeGAN2Demo
Videoデモ: huggingface
Pytorchモデル：bryandlee/animegan2-pytorch
Tensorflowオリジナルリポジトリ: TachibanaYoshino/AnimeGANv2
CoreML変換記事：AnimeGANv2をCore MLに変換してiOSでつかう【変換済みモデルあり】
🐣
フリーランスエンジニアです。
お仕事のご相談こちらまで
簡単な開発内容をお書き添えの上、お気軽にご連絡ください。
rockyshikoku@gmail.com
Core MLやARKitを使ったアプリを作っています。
機械学習／AR関連の情報を発信しています。