试玩 artline-demo

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

图片src：https://images.pexels.com/photos/2036646/pexels-photo-2036646.jpeg?cs=srgb&dl=pexels-anastasiya-gepp-2036646.jpg&fm=jpg

IU ღ( ´･ᴗ･` )比心 脸走形了。。 可以试着拿出铅笔和白纸来画一画了

image.png

IU21.jpg

注意

• 这里的模型输出结果的格式是固定的， 所以有些图片会出现被裁切了，不全；模型的learn predict结果就已经是长宽比是等高的，不清楚有没有办法调节，有清楚的希望可以告知下
• 跑这个还是很依赖电脑GPU资源的，个人的电脑跑一次后，就不能跑第二个图片了，出现CUDA Outof memory exception。
  尝试内存清理回收，没啥效果，只好重新运行开始跑了。

        # gc.collect()
        # torch.cuda.empty_cache()
        # time.sleep(10)