基于caffe的艺术迁移学习 style-transfer

Posted 2020-08-09 mitutao

这个是去年微博里面非常流行的，在git_hub上的代码是https://github.com/fzliu/style-transfer

比如这是梵高的画

这是你自己的画

然后你想生成这样

 

怎么实现呢在caffe上，很简单。

1 首先在 https://github.com/fzliu/style-transfer 把代码下载下来，另外主要这个代码基于pycaffe的，需要将pycaffe编译好。

最好是在电脑上装一个python progressbar包 

   在windows cmd下输入 pip install progressbar

2 解压style-transfer-master文件夹，运行\scripts 下的download_models.sh 就会下载模型，

在model文件夹下会出现caffenet、vgg16、vgg19、googlenet等文件夹，但是有时候model文件夹里面还是没有.model文件，一般来说.model文件有几十到几百M不能

我们可以打开download_models.sh，按照里面的链接直接下载。

3 运行代码

python style.py -s <style_image> -c <content_image> -m <model_name> -g 0

在windows下 切换到style.py 所在的目录，输入代码如下

然后就能运行，得到上述结果了。