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関連の情報を発信しています。

Twitter
Medium