AnimeGAN2をエクスポートします
手順
import torch
# Get the model from torch hub.
model = torch.hub.load("bryandlee/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1").eval()
# Convert.
dummy_input = torch.randn(1, 3, 512, 512)
input_names = [ "input" ]
output_names = [ "var_444" ]
torch.onnx.export(model, dummy_input, "animegan2_face_paint_512_v1.onnx", verbose=True, input_names=input_names, output_names=output_names, opset_version=11)
## Specify opset_version for upsampling layer.
🐣
フリーランスエンジニアです。
お仕事のご相談こちらまで
rockyshikoku@gmail.com
Core MLやARKitを使ったアプリを作っています。
機械学習／AR関連の情報を発信しています。