在Docker中运行torch版的neural style
2016-11-04 20:40
225 查看
相关的代码都在Github上,请参见我的Github,https://github.com/lijingpeng/deep-learning-notes
敬请多多关注哈~~~
torch版本的neural style来自于jcjohnson-neural-style,支持CPU和GPU,依赖torch7和loadcaffe,Docker环境中已经安装好了这些依赖。
neural style需要训练好的VGG网络结果,这里需要提前下载好:
需要加载以下两个文件:
VGG_ILSVRC_19_layers.caffemodel
VGG_ILSVRC_19_layers_deploy.prototxt
caffemodel文件比较大,建议使用下载工具下载到本地。
我这里直接把用户主文件夹映射进去了,实际可以根据自己文件的存放位置来调整。
-style_image 表示风格图片文件位置
-content_image 表示内容图片的位置,也就是你想要改变风格的文件
-output_image 表示输出文件位置
-model_file 表示下载好的caffemodel文件
-proto_file caffemodel模型的配置文件
-gpu -1 -1表示不采用GPU,使用CPU版本
-optimizer adam 优化方式选择adam,速度较快,但是结果一般没有L-BFGS好
-num_iterations 迭代次数
-print_iter 1 每一轮迭代都要在控制台上显示一次结果
更多参数设置请参考:neural-style
接下来就是漫长的等待了,如果使用CPU的话这个等待时间将会非常长…
最后在-output_image的指定位置查看模型的数据结果。需要注意的是请不要把模型的数据结果输出到Docker的内部文件夹中,因为一旦Docker执行过程中崩溃或者关闭,你的所有内容将会被丢弃。因此一定要把结果放到映射好的宿主机文件夹中。
敬请多多关注哈~~~
在Docker中运行torch版的neural style
TensorFlow neural-style, TensorFlow版本的实现比Torch版本的实现要慢很多,因此本文介绍如何运行torch版本的neural style。为了避免搭建环境时候解决各种烦人的依赖问题,这里依然使用Docker环境,Dockerfile来自这里. 环境搭建可以参考这里torch版本的neural style来自于jcjohnson-neural-style,支持CPU和GPU,依赖torch7和loadcaffe,Docker环境中已经安装好了这些依赖。
下载训练好的VGG网络
首先clone代码:https://github.com/jcjohnson/neural-style
neural style需要训练好的VGG网络结果,这里需要提前下载好:
sh models/download_models.sh
需要加载以下两个文件:
VGG_ILSVRC_19_layers.caffemodel
VGG_ILSVRC_19_layers_deploy.prototxt
caffemodel文件比较大,建议使用下载工具下载到本地。
在Docker中运行
第一步:运行Docker
由于VGG训练结果文件、图片文件等都存放到了本地电脑上,因此我们在启动docker时需要把这些文件映射到Docker中docker run -it -p 8888:8888 -p 6006:6006 -v /Users/frank:/root/sharedfolder floydhub/dl-docker-load:cpu
我这里直接把用户主文件夹映射进去了,实际可以根据自己文件的存放位置来调整。
第二步:进入neural style代码的目录
假设clone下来的代码存放到了/Users/frank/Downloads/neural-style下,则:cd ~/sharedfolder/Downloads/neural-style
第三步:执行
th neural_style.lua -style_image examples/inputs/starry_night.jpg -content_image ~/sharedfolder/Downloads/content.png -output_image ~/sharedfolder/Downloads/nn_out.png -model_file ~/sharedfolder/Downloads/VGG_ILSVRC_19_layers.caffemodel -proto_file ~/sharedfolder/Downloads/VGG_ILSVRC_19_layers_deploy.prototxt -gpu -1 -optimizer adam -num_iterations 800 -print_iter 1
-style_image 表示风格图片文件位置
-content_image 表示内容图片的位置,也就是你想要改变风格的文件
-output_image 表示输出文件位置
-model_file 表示下载好的caffemodel文件
-proto_file caffemodel模型的配置文件
-gpu -1 -1表示不采用GPU,使用CPU版本
-optimizer adam 优化方式选择adam,速度较快,但是结果一般没有L-BFGS好
-num_iterations 迭代次数
-print_iter 1 每一轮迭代都要在控制台上显示一次结果
更多参数设置请参考:neural-style
接下来就是漫长的等待了,如果使用CPU的话这个等待时间将会非常长…
最后在-output_image的指定位置查看模型的数据结果。需要注意的是请不要把模型的数据结果输出到Docker的内部文件夹中,因为一旦Docker执行过程中崩溃或者关闭,你的所有内容将会被丢弃。因此一定要把结果放到映射好的宿主机文件夹中。
相关文章推荐
- 在Docker中运行torch版的neural style
- docker运行mywebsql
- 使用docker构建jenkins镜像并运行容器
- Docker 容器的运行(八)
- 转:使用 Docker 搭建 Java Web 运行环境
- 使用 Docker 搭建 Java Web 运行环境
- 详解挂载运行的docker容器中如何挂载文件系统
- 使用Docker在Nginx上运行简单的HTML网页
- DOCKER 给运行中的容器添加映射端口
- 学会容器服务帮你打造Docker云端最佳运行环境
- docker ――查找与运行whalesay镜像
- 用 Deployment 运行应用 - 每天5分钟玩转 Docker 容器技术(123)
- 【Docker】Docker运行单机版Spark
- Docker搭建Javaweb运行环境
- docker上运行opengl依赖的服务
- 如何在windows 7运行docker
- Docker笔记一:基于Docker容器构建并运行 nginx + php + mysql ( mariadb ) 服务环境
- 使用 Docker 搭建 Java Web 运行环境
- 使用 Docker 搭建 Java Web 运行环境
- Docker运行Mysql实例