この記事は、data-refinement Advent Calendar 2018 一日目の記事です
概要
TensorFlowLite を試してみた記事です。
Codelabs の以下を、ためしました。
TensorFlow for Poets 2: TFLite Android
内容は
画像認識のモデルを、Tensorflow lite によって convert し、 android 実機もしくは、カメラアクセスを許可したエミュレータ上で実行するチュートリアルです
ここでは、振り返りと感想を記事にします。
Setup から コンバートまで
早速, チュートリアルでは、"tensorflow==1.7.*"
を指定していますが、
tflite_convert
が同梱されるのは、1.9以降とのことで、ここを、1.9.* に変更しました
1.7 では toco
でしたね
python の環境づくりに、時間を溶かされるのを避けるために、
docker image を作りました
docker image for TensorFlow for Poets 2
jupyter notebook もおいてあるので、コンバートファイル作成まで、サクサク行くはずです
Setup the Android app から 最後まで
Android Studio は、もうインストール済みですね?!
docker image に追加しようと思いましたが、ちょっと時間切れなので
あとで追加します
github 上のリポジトリに
サンプルアプリの android プロジェクトがあるので、それをビルドします
学習済みモデルをコピーする前では、画像認識のいい結果が得られないのですが
コンバート済みの学習済みモデルを、コピーするだけで、いい感じに画像認識してくれます
How does it work?
TensorFlow Lite コードの特徴... 追記します