的PyTorch实现
从原始存储库转换重量(需要TensorFlow 1.x)
git clone https://github.com/TachibanaYoshino/AnimeGANv2
python convert_weights.py
推理
python test.py --input_dir [image_folder_path] --device [cpu/cuda]
转换后的[ ]样式模型的结果
(输入图像,原始tensorflow结果,pytorch结果从左到右)
脸部模型:从此提取出来,带有L2 + VGG + GAN损失和CelebA-HQ图像。 有关推断,请参见test_faces.ipynb 。 可以从下载模型文件(8MB)。 享受!
注意:不包含培训代码/在RTX3090 + PyTorch1.7.1上进行了测试/由于,转换后的权重结果略有不同
1