WindowsでMXNet-Gluon-Style-Transferを使う(GPUあり)

CUDA9.1→9.2にアップデート
2018年8月11日動作確認

環境

Windows10 Pro 64bit
CUDA9.2
cudnn7.2.1

Anacondaで仮想環境を作成

conda create -n mxnet python=3.6 anaconda
activate mxnet
  • pipのアップデート
python -m pip install --upgrade pip
  • msgpackのインストール
pip install msgpack

mxnetのインストール

pip install mxnet-cu92==1.3.0b20180810

MXNetのソースをダウンロード

こちら』から最新のソースコードをダウンロードして解凍

imageのダウンロード

Linuxsvnコマンドが含まれるため、bash on Windowsを使用しないといけない
まずはsubversionのインストール

sudo apt-get install subversion

続いて、『incubator-mxnet-master/example/gluon/style_transfer』フォルダ内に移動して以下を実行

python3 download_images.py

学習済みモデルのダウンロード

こちらは、通常のWindowsから実行可能
『incubator-mxnet-master/example/gluon/style_transfer』フォルダ内に移動して以下を実行

python models/download_model.py

実行

python main.py eval --content-image images/content/venice-boat.jpg --style-image images/styles/wave.jpg --model models/21styles.params --content-size 1024 --cuda 1