Animegan2-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]
样品
注意:由于不包括训练代码/双线性上采样问题,转换权重的结果略有不同。