
AnimeGANv2-PyTorch: AnimeGANv2的PyTorch版本实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
AnimeGANv2-PyTorch是基于PyTorch框架实现的AnimeGANv2代码库,用于将照片风格转换成日式漫画风格,提供模型训练和预处理工具。
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文件。
注意:不包含训练代码,在RTX3090 + PyTorch1.7.1上进行了测试,由于转换后的权重结果略有不同。
全部评论 (0)
还没有任何评论哟~


