Windowsでneural-styleをやってみた

windowsでもDeepLearningとやらを試してみたいということで、Ostagramで一部話題のneural-styleを入れてみた。普通に入れるのは難しそうなので、Dockerを使って動かしてみた。

動作環境

OS：Windows10

CPU:Core i3-3217U 1.8GHz

Memory:8G

ちなみにCPUはこれだと画像一枚8時間位かかりました。メモリーは、8Gあったほうがいいような気がします。

手順

以下を参考にdockerを入れる
https://docs.docker.com/windows/

CPU、メモリーをチューニングしてみる。（自分の環境に合わせてね）

docker-machine stop
docker-machine rm default
docker-machine create -d virtualbox --virtualbox-memory 4096 --virtualbox-cpu-count 4 default

環境構築

docker pull kchentw/neural-style

docker run -it -m 4g --cpuset-cpus -v /c/Users/hoge/work:/root/work kchentw/neural-style bash

上は、windows上のc:\Users\hoge\workに写真を置く前提です。

 実行

cd /root/neural-style

th neural_style.lua -gpu -1 -style_image <style.jpg> -content_image  <content.jpg>

 実行ディレクトリの配下にout.jpgができるので、cp out.png ../work/とかして、windowsから取得する。

一応動いてますが、非常に遅い。どうやら１コアしか使われていないみたい。チューニングでなんとかなるのかな？