最近看到一个非常有趣的照片风格迁移的工作,并且是可以上手把玩的,将自己的照片转换为漫画风格。
量子位也花了大篇幅的报道,效果惊人啊,据说新海诚导演就转发过:
又到了光说不练假把式的时候了,这么有趣的内容怎么能不上手试一试呢?(其实这根本不是最近才有的项目,不知道为啥又被炒起来了。但是很有趣,不是吗?)
量子位也提供了3种方式,一种是抱抱脸的在线版,亲测无效!什么要等几个小时,上传图片就失败了。第二种就是Colab版本,这也是随便可以玩的方法吗?放弃。看来只能是第三种,老老实实去github上下载原项目了。
2.1. 下载原项目但是原项目是这样给的指南,我只能无力吐槽,没有requirements.txt,也没有开箱即用,这看来得看看源码了。
首先,我们不需要再去做第一个转换模型的 *** 作了,因为它已经将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的图片大小,否则就会爆显存。正如原文提醒的那样,为了保证照片的质量要高,建议用大头贴的形式,尽可能细致的保存面部特征。
很多同学也反映,这玩意很多美图APP上面都有,费这么大劲干啥。甚至好几个同学说,他们更想自己画。其实吧,这个项目可能最重要的是,离线版的隐私泄露会更少一些吧,我只能这样劝慰我自己。
你是不是也想试一下了?
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)