AKIBAko

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

RaspberryPi に Caffe

RaspberryPi で画像認識なんかをさせてみようと Caffe をいれてみた
学習は無理でもインファレンスはできるんじゃないかと。。

http://caffe.berkeleyvision.org/install_apt.html
ここを参考に Raspberry Pi で下記を実施  

  • General dependencies
$ sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
$ sudo apt-get install --no-install-recommends libboost-all-dev

 

  • BLASインストール
$ sudo apt-get install libatlas-base-dev 

 

  • その他パッケージ
$ sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

 

  • Caffe を取得
$ git clone https://github.com/BVLC/caffe.git
$ cd caffe

 

  • Caffe のビルド
    CPU_ONLY にしなければ、RaspberryPi でも GPU を使ってくれるのかな?
$ mkdir build
$ cmake ..
$ make all
$ make instll
$ make runtest

 

  • pycaffe のビルド
    python で使いたいので pycaffe をビルド
$ make pycaffe

 

  • caffe へのパス設定
$ cd ~/
$ echo 'export PYTHONPATH=~/caffe/python/:$PYTHONPATH' >> ~/.bashrc
$ source .bashrc

 

パスが設定できたところでインストール完了。 あとは python で caffe がインポートできるか確認してみる

ImportError: No module named skimage.io

skimage が無いとのこと。とりあえず apt-get でインストール

$ sudo apt-get install python-skimage

再度、caffe インポートすると、次は protobuf が無いらしい

ImportError: No module named google.protobuf.internal

protobuf も apt-get でインストールすると、やっと caffe のインポートに成功

pi@raspberrypi:~ $ python
Python 2.7.9 (default, Sep 17 2016, 20:26:04)
[GCC 4.9.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>>
>>> import caffe
>>> caffe.__version__
'1.0.0-rc3'

とにかくビルドにめちゃ時間かかったので、動きは今度確認しよう