Movidiusでラズパイでも画像認識高速化が可能に

こんにちは。

AI coordinatorの清水秀樹です。

前回記事でも紹介したラズパイでSSD_Kerasを試した方法に続き、今回は兼ねてからチャレンジしてみたかったMovidiusを使った画像認識の高速化にチャレンジしてみたいと思います。

Raspberry PiでSSD_Kerasを使った物体検出にチャレンジ！！

Movidiusはintelが開発したUSB型のDeepLearning用開発ツールです。

「モビディウス」と読むようです。

当記事は、ラズパイ上でMovidiusを使ったリアルタイム画像認識ができるところまでを紹介します。

Movidiusは安価でラズパイの高速化が可能になるDeepLearningツールなので、興味がある方は是非チャレンジしてみて下さい。

Movidiusをインストールする前の注意点として、microSDカードの容量は３２GB以上にしましょう。

１６GBでは領域が足らずにインストールできません。

しかもインストールの最後の最後でエラーとなるので、それまでインストール作業に費やしてきた時間（約５時間）が無駄になります。

筆者は２回やられました。

ケチらずに３２GB以上を準備しましょう。

Swapメモリの増強から

Movidiusを使えるようにするためのインストール作業をスムーズに進められるようにするため、Swapメモリの増強から始めましょう。

これを実施しないと、ほぼ１００％の確率でインストール中にラズパイがフリーズします。

必ず実施しましょう。

tensorflowのインストール

続いて、以下のコマンドでtensorflowをインストールしましょう。

$ wget https://github.com/lhelontra/tensorflow-on-arm/releases/download/v1.3.1/tensorflow-1.3.1-cp35-none-linux_armv7l.whl
$ sudo pip3 install tensorflow-1.3.1-cp35-none-linux_armv7l.whl

１０分もあればインストールできます。

Movidiusのセットアップ

公式サイトのインストール手順に沿ってインストールしましょう。

非常に簡単にインストールできますが、５時間くらいかかりました。

以下のコマンドでラズパイ上にインストールできます。

mkdir -p ~/workspace
cd ~/workspace
git clone https://github.com/movidius/ncsdk.git
cd ~/workspace/ncsdk
make install

make installが完了したら、新しくターミナルを起動して以下のコマンドを入力します。

cd ~/workspace/ncsdk
make examples

このインストールが完了するとOpenCVも同時インストールされます。

ラズパイにOpenCVをインストールする作業は依存関係も含め非常に困難ですが、MovidiusのセットアップはこのOpenCVのインストールも同時にやってくれるので非常にありがたいですね。

OpenCVだけでなく、NumPy、SciPy、scikit-learnも合わせてインストールしてくれます。

繰り返しになりますが、この作業は５時間くらいかかります。

寝る直前に、make examples することをオススメします。

ちなみに後から知ったのですが、以下のコマンドなら一発で全てインストールできます。

こちらの方が楽ですね。

$ git clone http://github.com/Movidius/ncsdk && cd ncsdk && make install && make examples

サンプルの実行

インストールが完了したら、早速サンプルプログラムを動かしてみましょう。

（注意：以下のコマンドはインストール時とディレクトリが違うため注意してください。）

pi@raspberrypi:~/ncsdk/examples/tensorflow/inception_v3 $ python3 run.py
Number of categories: 1001
Start download to NCS...
*******************************************************************************
inception-v3 on NCS
*******************************************************************************
547 electric guitar 0.98828
403 acoustic guitar 0.0077209
715 pick, plectrum, plectron 0.0015087
421 banjo 0.00092602
820 stage 0.00065947
*******************************************************************************
Finished

何やら、ギターだと結果が返ってきます。

ソースコードを覗くと参照している画像が分かりました。

なるほどなるほど。

カメラ映像からの画像認識

それでは最終目標であるリアルタイム映像からの画像認識にチャレンジしてみたいと思います。

ソースはこちらのサイトに紹介されているソースを使用させて頂きました。

ありがとうございます。

実行結果は見事に映像からの画像認識ができました。

ただ確かに画像認識スピードは早いのですが、あくまでラズパイ単体に比べて圧倒的に早いというだけで、画像の遅れは３秒ぐらいあります。

Movidiusを複数繋げて動かすこともできるようなので、さらなる高速化を目指す場合はそういった手段を検討してみるのも手かもしれませんね。

それでまた。