読者です 読者をやめる 読者になる 読者になる

AKIBAko

自分のためのナレッジ的ブログ

DIGITS で MNIST リベンジ

Raspberry pi に DIGITS をインストールするというあまりにも無駄なことに頑張ってみた

DIGITS インストール

DIGITS/BuildDigits.md at master · NVIDIA/DIGITS · GitHub の手順どおりで特にエラーもなく DIGITS 起動までOK

関連ライブラリのインストール

$ sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-pip python-protobuf python-scipy

DIGITSダウンロード

$ DIGITS_ROOT=~/digits
$ git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT

Pythonパッケージをインストール

$ sudo pip install -r $DIGITS_ROOT/requirements.txt
$ sudo pip install -e $DIGITS_ROOT

DIGITS起動

$ cd ~/digits
$ ./digits-devserver

ちょっと感動。。Raspberry pi でも DIGITS が動いた f:id:minori-san:20161117000735p:plain  

MNIST トレーニング

データセットを取得

$ cd ~/digits
$ python -m digits.download_data mnist ~/mnist

データベースを作成してトレーニングを開始

f:id:minori-san:20161117001543p:plain

トレーニングは開始できたが、絶望的に時間がかかる。
ラズパイがんばれ!とりあえず2エポックまわしたときのモデルで推論

f:id:minori-san:20161117001941p:plain

f:id:minori-san:20161117001949p:plain

おー、ちゃんと分類できている。すばらしい