这里不是教你如何生成可以实现风格迁移的模型(我们以后可以教),这里是借用别人训练好的模型和代码,自己跑着玩。

上图就是一只猫,加一幅著名浮世绘的话,生成一只浮世绘版猫咪。

我假设你已经有python环境(推荐py3),这个玩具需要下列库:

$ pip3 install scipy pillow tensorflow

克隆大神代码:

$ git clone https://github.com/lengstrom/fast-style-transfer.git

进入 fast-style-transfer 目录下载一个你喜欢的模型,例如 rain_princess.ckpt。就可以跑代码玩起来了:

python evaluate.py --checkpoint ./rain-princess.ckpt --in-path <path_to_input_file> --out-path ./output_image.jpg

切勿 transfer 太大的图像,你可能要等很久!


附录:几个style的样式:

  • la muse

  • rain princess

  • the scream

  • the shipwreck of the minotaur

  • udnie