0
点赞
收藏
分享

微信扫一扫

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

龙毓七七 2022-03-15 阅读 85

Python 超简单实现9种图像风格迁移_python


图像风格迁移已经属于比较成熟的领域了,现在连实时的风格迁移都不成问题。之前一直想出一篇这样的文章,但无奈于大部分开源项目配置起来非常麻烦,比如 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训练好的模型对下面的图片做一次风格迁移。


Python 超简单实现9种图像风格迁移_灰度值_02


1.选择模型


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

Python 超简单实现9种图像风格迁移_像素点_03

candy

Python 超简单实现9种图像风格迁移_python_04

mosaic


Python 超简单实现9种图像风格迁移_灰度值_05

starry_night


Python 超简单实现9种图像风格迁移_灰度值_06

udnie


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

2.克隆OpenCV源码

我们直接克隆OpenCV开源项目中关于DNN图像迁移的例子,地址是:
​​​​​​

代码:

Python 超简单实现9种图像风格迁移_python_07


注意,源代码是基于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

效果:

Python 超简单实现9种图像风格迁移_灰度值_08


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

Python 超简单实现9种图像风格迁移_python_09

Python 超简单实现9种图像风格迁移_像素点_10

Python 超简单实现9种图像风格迁移_灰度值_11


如果你喜欢今天的Python 教程,请持续关注Python实用宝典

Python实用宝典 (pythondict.com)

不只是一个宝典

Python 超简单实现9种图像风格迁移_python_12

举报

相关推荐

0 条评论