次のコード (tf_lite_converter.py) を使用して、.pb ファイルに保存されている既存の凍結グラフを変換しようとしました。
#!/usr/bin/env python
import sys
import tensorflow as tf
from tf.contrib.lite import convert_savedmodel
convert_savedmodel.convert(
saved_model_dir="/frozen_inference_graph.pb",
output_tflite="/TF_Lite_Model")
でコードを実行すると
python tf_lite_converter.py
私のアナコンダ環境では、エラーが発生します:
ImportError: No module named tf.contrib.lite
私の目標は、.pb-graph から tensorflowlite-model を取得して、Android アプリケーションで使用することです。すでに bazel を介して toco を使用して tflite をビルドしようとしましたが、おそらく (またはほとんどの場合) 何か間違ったことをしたか、問題を修正する正しい方法ではありませんでした。
Tensorflow-Developers からの参照ビデオ: https://youtu.be/FAMfy7izB6A?t=11m49s