漫画版的你，离线版AnimeGANv2初体验

Posted 2021-12-17 刘炫320

1. 前言

最近看到一个非常有趣的照片风格迁移的工作，并且是可以上手把玩的，将自己的照片转换为漫画风格。
量子位也花了大篇幅的报道，效果惊人啊，据说新海诚导演就转发过：


又到了光说不练假把式的时候了，这么有趣的内容怎么能不上手试一试呢？（其实这根本不是最近才有的项目，不知道为啥又被炒起来了。但是很有趣，不是吗？）

2. 实操

量子位也提供了3种方式，一种是抱抱脸的在线版，亲测无效！什么要等几个小时，上传图片就失败了。第二种就是Colab版本，这也是随便可以玩的方法吗？放弃。看来只能是第三种，老老实实去github上下载原项目了。

2.1. 下载原项目

但是原项目是这样给的指南，我只能无力吐槽，没有requirements.txt，也没有开箱即用，这看来得看看源码了。

2.2. 修改内容

首先，我们不需要再去做第一个转换模型的操作了，因为它已经将pytorch版本的模型集成到weights里了，你可以在参数–checkpoint里指定模型，自带了4个模型，比如所谓的V2版本模型就可以这样执行：

python3 test.py --checkpoint ./weights/face_paint_512_v2.pt

系统默认的输入文件夹放在./samples/inputs里，输入文件夹放在./samples/results里。

其他的就不用再去做什么改动了。

2.3 安装环境

环境要求倒是挺低的，稍微新一些的pytorch，比如1.8.0的torch和3.8的python环境就可以跑起来。唯一需要额外安装的就是cv2这个包，这个包是啥呢，其实是个opencv的包。根据报错信息，我们查找到是使用pip3 install opencv-python命令来安装即可。

2.4 实战结果

亲身实践的效果如下：
原图

模型paprika.pt

模型face_paint_512_v2.pt

效果看起来还不错吧，不过唯一的缺点，图不能太大，即使是24G的3090，也只能承受600*800的图片大小，否则就会爆显存。正如原文提醒的那样，为了保证照片的质量要高，建议用大头贴的形式，尽可能细致的保存面部特征。

4. 小结

很多同学也反映，这玩意很多美图APP上面都有，费这么大劲干啥。甚至好几个同学说，他们更想自己画。其实吧，这个项目可能最重要的是，离线版的隐私泄露会更少一些吧，我只能这样劝慰我自己。

你是不是也想试一下了？