Jetson nanoでLobeから出力したTensorFlow Liteのモデルを動かす

1 min read読了の目安(約1400字

Getting Started With Jetson Nano Developer Kit | NVIDIA DeveloperにしたがってJetson nanoのUbuntuにログインできるようにする。

onnxruntime_gpu を入手してインストールします

# Download pip wheel from location mentioned above
$ wget https://nvidia.box.com/shared/static/49fzcqa1g4oblwxr3ikmuvhuaprqyxb7.whl -O onnxruntime_gpu-1.6.0-cp36-cp36m-linux_aarch64.whl

# Install pip wheel
$ pip3 install onnxruntime_gpu-1.6.0-cp36-cp36m-linux_aarch64.whl

TensorFlow1系をインストール

Lobeが出力するモデルが1.x系用なのでそちらをインストールします。下記のページのインストール方法に従い、TensorFlowをいれるときだけ1.xを指定します。

Installing TensorFlow For Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

1.Xを指定するコマンド(3. Installing TensorFlowにあります)

$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 ‘tensorflow<2’

TensorFlow Liteのランタイムを入れる

pip3 install https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp36-cp36m-linux_aarch64.whl

いれてないとこういうエラーが出る

ModuleNotFoundError: No module named 'tflite_runtime'

lobeのモデルを動かす

lobeからTensorFlow Lite形式でエクスポートするとサンプルファイルがあるので、その引数に画像を渡して実行

pip3 install -r requirements.txt #必要なライブラリをインストール
python3 tflite_example.py '../dokin.jpg'  #予測
{'Confidences': [0.009673559106886387, 0.990326464176178, 2.5323764418772043e-08], 'Prediction': 'Dokin-chan'}