CNN之绘画风格迁移-附源码地址

废话不说：先上成果！

成果1（保留原本原色）

成果2（保留原本颜色）

成果3（丢弃原本颜色）

👉源码地址👈

操作步骤

• 先从git上打包下载所需的代码文件，里面包含各种案例，安装环境，源码py的文件。

  

• 通过pip install -r requirements.txt，安装相关的运行环境，其中requirements.txt存放在打包下载好的文件夹中。

• 在git地址下，有相应的预训练好的模型，同样需要下载。然后存放在原先打包下载好的文件夹内。
  

• 打开cmd命令行，cd 到相应下载好的文件夹内。
  

• 在同样的命令行内运行py文件，同时指定输入需要转换的图片content，迁移风格的图片styles，以及输出的图片output，然后再回车运行即可。如下图，我这边输入了123.jpg作为content，2-style1.jpg作为迁移风格，321.jpg作为输出图片。大家可以根据自身情况，对这三个参数进行修改。
  

• 我这边使用的是1650的显卡，训练一张图片大概在2分钟左右。但是CPU却要数十分钟，大家最好使用GPU进行操作。