利用已有Caffemodel进行风格迁移

1 

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

注意：编译pycaffe的时候可能会遇到提示缺少Python27_d.lib，解决方案参考以下博文，https://blog.csdn.net/kele_imon/article/details/79137876

可以完美解决，若有问题请留言继续沟通。

 

。代码包如图：

最好是在电脑上装一个python progressbar包 ，在windows cmd下输入 pip install progressbar

运行情况如图所示：

备注：输入pip install progressbar的时候需要等待以下，就会继续运行，运行成功会有如上截图。

2 

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

3 

在style-transfer-master文件夹下打开cmd，输入如下指令：

代码格式：python style.py -s <style_image> -c <content_image> -m <model_name> -g 0

python style.py -s images/style/starry_night.jpg -c images/content/nanjing.jpg -m vgg19 -g 0

结果报错，错误如下图所示：

 

具体错误描述如下：

WARNING：Logging before InitGooleLogging()is written to STDERR

Cannot use GPU in COU-only Caffe:check mode.

 

解决办法：我的caffe没有装GPU版本，而是Windows-only-cpu所以是这个原因，然后我把运行脚

 

本代码修改如下：

python style.py -s images/style/starry_night.jpg -c images/content/nanjing.jpg -m vgg19 -g 0

 

修改为：python style.py -s images/style/starry_night.jpg -c images/content/nanjing.jpg -m vgg19 -g -1

 

修改原因：可见上面例子中，style-image对应starry_night.jpg 即风格图像, content_image对应自己的照片nanjing.jpg, 模型选择vgg19, g 0对应选择默认的GPU，如果是g -1则为CPU

结果得到如下错误：

 

错误描述：error:argument -g/--gpu-id:invaid int value:’0python’

 

解决办法：居然是我复制代码错误。。。。修改结束后

 

回车后得到如下结果：

 

https://blog.csdn.net/wnma3mz/article/details/77158967

以上网页显示了caffe和tensorflow的风格迁移步骤，上图得到三条结果后没有加载过程，不知道为什么，我还在查找原因。

 

结果是因为没有下载caffemodel文件，然后打开

这个文件夹里面有一个叫做download_models.sh的文件，用vs打开如下，

 

然后带点击对于模型链接可直接下载caffemodel文件，放在对应的models中的网络文件中，比如我下载了VGG19，效果如下图：

 

然后再次在style-transfer-master文件夹中cmd运行上述代码，得到如下错误：（本不应该出现如下错误）：其实不是错误继续等待即可，

 

 继续等待之后出现如下图：

 

表示训练结束，在outputs文件夹中出现所生成的图片：

，表示实验结束。

 

 

附录

检查pycaffe是否可用，来到caffe文件夹下的Python文件夹（./caffe/python），打开python 然后：import caffe

有报错：

如果是“no module caffe”的报错，修改一下环境变量：

 

下图表示pycaffe可用