python图像风格迁移_Python超简单实现9种图像风格迁移

图像风格迁移已经属于比较成熟的领域了，现在连实时的风格迁移都不成问题。之前一直想出一篇这样的文章，但无奈于大部分开源项目配置起来非常麻烦，比如 luanfujun/deep-photo-styletransfer 项目，需要安装 CUDA、pytorch、cudnn等等，配置完一天都过去了。

不过最近我发现一个非常好的开源应用项目，那就是OpenCV的DNN图像风格迁移。你只需要安装OpenCV就可以使用，在cmd/terminal中输入(如果你还没有安装Python，请看这篇文章 《Python安装》)：

pip install python-opencv

不过它也有局限性，我们只能用别人训练好的模型进行风格迁移，如果我们要自定义风格，那就必须配置cudn等工具，使用 deep-photo-styletransfer 等项目的方法进行训练，今天的教程我们拿fast-neural-style训练好的模型对下面的图片做一次风格迁移。

1.选择模型

fast-neural-style放出的模型风格一共有9种，我们将一一尝试，其中部分风格如下比如：

candy

mosaic

starry_night

udnie

模型文件可以扫描下方二维码关注 Python实用宝典，回复风格迁移下载，里面有全部9个模型风格的资源和源代码。

2.克隆OpenCV源码

我们直接克隆OpenCV开源项目中关于DNN图像迁移的例子，地址是：https://github.com/opencv/opencv/blob/3.4.0/samples/dnn/fast_neural_style.py

代码：

注意，源代码是基于Python2的，所以第46行少了括号，如果你是Python3请注意补上括号。这份代码可以直接使用， parser 里定义了5个参数，--input输入要迁移的图像宽度和高度， median_filter 是中值滤波器， 基本思想是用像素点邻域灰度值的中值来代替该像素点的灰度值，因此理论上数值越大，图像越平滑，输出的结果细节越好(不确定)。亲自试了一下median_filter对图像的影响，发现改变微乎其微，因此直接为默认值即可。

3.开始迁移

将第二步的代码保存到一个文件中，命名为1.py，在CMD/Terminal中带参数运行脚本，其中input是源图像路径，model是迁移的风格模型文件，如运行：

$ python 1.py --input 1.jpg --model udnie.t7

效果：

全部9种风格的迁移效果：

如果你喜欢今天的Python 教程，请持续关注Python实用宝典，如果对你有帮助，麻烦在下面点一个赞/在看

，有任何问题都可以在下方留言，我们会耐心解答的！

源代码和模型文件可以关注我们下方公众号 Python实用宝典，回复风格迁移下载，里面有全部9个模型风格的资源和源代码。

点击下方阅读原文可以获取所有代码和原图哦！

Python实用宝典 (pythondict.com)

不只是一个宝典

欢迎关注公众号：Python实用宝典