animegan2-pytorch:AnimeGANv2的PyTorch实现

时间:2024-05-25 04:40:38
【文件属性】:

文件名称:animegan2-pytorch:AnimeGANv2的PyTorch实现

文件大小:7.37MB

文件格式:ZIP

更新时间:2024-05-25 04:40:38

style-transfer image2image JupyterNotebook

的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上进行了测试/由于,转换后的权重结果略有不同


【文件预览】:
animegan2-pytorch-main
----test.py(2KB)
----test_faces.ipynb(1.26MB)
----convert_weights.py(4KB)
----samples()
--------faces()
--------inputs()
--------face_results.jpg(843KB)
--------compare()
----model.py(4KB)
----.gitignore(2KB)
----README.md(1KB)

网友评论