`
m635674608
  • 浏览: 5090947 次
  • 性别: Icon_minigender_1
  • 来自: 南京
社区版块
存档分类
最新评论

在Docker中运行torch版的neural style

 
阅读更多

相关的代码都在Github上,请参见我的Github,https://github.com/lijingpeng/deep-learning-notes
敬请多多关注哈~~~

在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执行过程中崩溃或者关闭,你的所有内容将会被丢弃。因此一定要把结果放到映射好的宿主机文件夹中。

 

http://www.cnblogs.com/lijingpeng/p/6031634.html

http://www.cnblogs.com/lijingpeng/p/6009476.html

http://blog.csdn.net/lijingpengchina/article/details/53039051

分享到:
评论

相关推荐

    docker在旧版本windows的安装包

    在Docker早期版本中,它主要在Linux系统上运行,但随着技术发展和市场需求,Docker也推出了可以在Windows和Mac系统上运行的版本。Windows用户可以通过Docker for Windows来使用Docker的功能。不过,由于Windows系统...

    Docker中运行macOS:Docker-OSX

    推荐理由:一个让你在Docker中运行macOS操作系统的项目。通过仅需两条命令,用户可以在Docker容器中轻松启动一个macOS操作系统。这为开发者提供了一种简单而刺激的方式,让他们可以在非苹果硬件上体验macOS,从而...

    Ubuntu docker 20.10.12版本和docker-compose离线安装包

    Docker CE(Community Edition)是Docker的社区版,它包括了Docker引擎、容器运行时环境containerd以及用于与Docker进行交互的CLI(命令行界面)工具。在这个压缩包中,有三个与Docker相关的DEB文件: 1. `docker-...

    docker免安装版.tar.gz

    【标题】"docker免安装版.tar.gz"是一个包含Docker免安装版本的压缩包,它提供了无需正式安装即可运行Docker服务的便利性。这个版本适用于那些希望快速部署Docker环境,或者在不支持标准安装的系统上使用Docker的...

    docker文档中文.pdf

    docker从入门到精通,docker文档中文版。适合初学者了解docker各方面的基础知识。docker从入门到精通,docker文档中文版。适合初学者了解docker各方面的基础知识。docker从入门到精通,docker文档中文版。适合初学者...

    docker for mac 历史版本适用inter 10.15以下

    docker for mac 历史版本适用inter 10.15以下

    arm 架构 docker运行nginx镜像包

    arm 架构 docker运行nginx镜像包,arm 架构 docker运行nginx镜像包,arm 架构 docker运行nginx镜像包,arm 架构 docker运行nginx镜像包,arm 架构 docker运行nginx镜像包,arm 架构 docker运行nginx镜像包,arm 架构...

    Windows版本Docker桌面版安装包

    Docker Desktop 是一款强大的开发工具,使开发者能够在 Windows 环境中高效构建、共享和运行容器化应用程序。它提供了一种无缝的开发体验,允许用户通过简单的图形界面和命令行工具来管理 Docker 容器和镜像。 ...

    Docker windows版本

    要先用git里面的sh,就可以正常使用linux模拟环境,在用...Docker windows版本Docker windows版本Docker windows版本Docker windows版本Docker windows版本Docker windows版本Docker windows版本Docker windows版本

    torch-docker:Dockerfile 为 Torch7 创建镜像

    火炬码头 Dockerfile 为 Torch7 创建镜像

    Docker Desktop Installer 4.23.0

    Docker的4.23.0版本作为当前可用的稳定版本,提供了安装工具,供用户下载并安装到Windows系统上,确保了用户可以获取到最新且稳定的Docker运行环境,而不必担心旧版本的不兼容或已停止维护的问题。这一版本的安装...

    4.4.4版本windows-docker.zip

    这个版本可能是针对4.4.4版Docker Engine进行优化和更新的,确保了在Windows环境下的稳定性和兼容性。 【描述】"4.4.4版本windows-docker.zip"是一个压缩包,包含了运行和管理Docker所需的所有组件,如Docker引擎、...

    windows下安装docker

    值得注意的是,在安装过程中可能会提示是否创建桌面快捷方式以及是否将 Docker 添加到系统的 PATH 环境变量中。建议选择这些选项以便于后续操作。 **1.3 安装 VirtualBox** 由于 Docker Toolbox 需要在 VirtualBox...

    centos7操作系统下安装docker,及docker常用命令、在docker中运行nginx示例

    在本文中,建议的仓库地址是阿里云的镜像站点,通过yum-config-manager命令来添加Docker CE的仓库。完成仓库的设置后,接下来的操作是安装Docker Engine-Community,这里使用yum install命令安装docker-ce、docker-...

    docker-chromium:在 docker 中运行Chrome

    如果您想知道如何在 Docker 中运行 Chromium。 这个 docker image 可能是一个解决方案。 在 Docker 中运行 Chrome 的需要有助于部署测试。 例如,您可能有另一个 docker 镜像正在运行,您的应用程序配置为使用特定的...

    将Spring boot应用打包成镜像放到docker中运行

    将Spring boot应用打包成镜像放到docker中运行: https://blog.csdn.net/anla_/article/details/79943251

    Docker中文指南.zip

    在Docker中运行应用 2.3. 使用容器 2.4. 使用docker镜像 2.5. 连接容器 2.6. 管理容器数据 2.7. 使用Docker Hub ...

    docker&docker-compose 19.03.14二进制安装

    在IT行业中,Docker和Docker Compose是两个极为重要的工具,它们在容器化应用程序部署和管理方面扮演着核心角色。本文将详细讲解如何通过二进制方式安装Docker 19.03.14及其配套的Docker Compose,并讨论它们的主要...

    liunx系统docker离线升级

    在Linux系统中,Docker是广泛应用的容器化技术,它允许开发者将应用程序及其依赖打包成容器,以便在任何兼容的Linux环境下轻松部署和运行。然而,在没有网络连接的情况下,对Docker进行升级可能会变得具有挑战性。...

    activator-akka-docker, 如何在 Docker 中运行akka群集.zip

    activator-akka-docker, 如何在 Docker 中运行akka群集 Akka & Docker有关详细描述,请阅读这里博客条目。sbt docker:publishLocaldocker run --name seed-1 akka-docker:2.3.4 --seeddoc

Global site tag (gtag.js) - Google Analytics