利用已有Caffemodel进行风格迁移

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kele_imon/article/details/80694426

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的时候需要等待以下,就会继续运行,运行成功会有如上截图。

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

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

结果报错,错误如下图所示:

 

具体错误描述如下:

WARNINGLogging 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

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

 

结果是因为没有下载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可用 

 

猜你喜欢

转载自blog.csdn.net/kele_imon/article/details/80694426