「PyTorch Implementation of AnimeGANv2」のPythonでのONNX推論サンプル
PyTorch Implementation of AnimeGANv2のPythonでのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。
変換自体を試したい方はColaboratoryなどでAnimeGANv2_Convert2ONNX.ipynbを使用ください。
GeForce GTX 1050 Ti:約290ms
GeForce RTX 3060:約120ms
デモの実行方法は以下です。
python sample_onnx.py
応用例です。
高橋かずひと(https://twitter.com/KzhtTkhs)
AnimeGANv2-ONNX-Sample is under MIT License.
女性の画像はフリー素材ぱくたそ様の写真を利用しています。