artline是一种将图片转换为手绘风格的AI模型。
亲自试验本地玩了下，遇到点小问题，也踩过坑，已解决。
按照后面说的来操作，可以玩一玩咯。
原仓基础上修改了一点点：https://github.com/yyFFans/artline-demo
1. 克隆下载
- 访问仓下载源码
- 下载realease tag中的数据训练模型 ArtLine_650.pkl
访问下载慢，可以看看这个：解决github release下载慢的问题（含github下载提速方法）
2. 安装：
- python3.x
- 依赖包 (因为pytroch版本前后是不兼容的，所以必须要如下面这样来安装torch包)
pip install torch==1.6.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install -r requirements.txt
依赖包安装 建议 创建虚拟的env环境来单独安装，避免以后可能与其他项目相冲突，参考如下:
![]()
3. 跑起来看看效果
python app.py 或者直接pycharm里面run起来
image.png
artline-demo01.gif
IU ღ( ´･ᴗ･` )比心 脸走形了。。 可以试着拿出铅笔和白纸来画一画了
image.png
IU21.jpg
注意
- 这里的模型输出结果的格式是固定的， 所以有些图片会出现被裁切了，不全；模型的learn predict结果就已经是长宽比是等高的，不清楚有没有办法调节，有清楚的希望可以告知下
- 跑这个还是很依赖电脑GPU资源的，个人的电脑跑一次后，就不能跑第二个图片了，出现CUDA Outof memory exception。
尝试内存清理回收，没啥效果，只好重新运行开始跑了。# gc.collect() # torch.cuda.empty_cache() # time.sleep(10)
