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 が動いた 　

MNIST トレーニング

データセットを取得

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

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

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

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