animeganv2

AnimeGanV3

Input --> Output

Convert

pt --> TorchScript --> pnnx --> ncnnOptimize --> ncnn

import os
import torch
# 0. pt模型下载及初始化
model = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v2")
# model = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="paprika")
model.eval()

# 1. pt-->torchscript
traced_script_module = torch.jit.trace(model, torch.randn(1, 3, 512, 512))
traced_script_module.save("ts.pt")

# 2. ts --> pnnx --> ncnn
os.system("pnnx ts.pt inputshape=[1,3,512,512]")    # 2022年5月25日起，pnnx默认自动量化，不需要再次optmize

Example project

• Android: nihui/ncnn-android-styletransfer
• Desktop: baiyue/animegan_gui

Reference

• bryandlee/animegan2-pytorch