如何将TensorFlow JSON图形模型转换为.tflite?

时间:2019-07-17 22:30:59

标签: android json tensorflow model tensorflow-lite

我正在将this TensorFlow JS tutorial移植到Android。我必须将this JSON graph TensorFlow model转换为.tflite以便可以在其中使用。

我在Google上搜索了TensorFlow模型转换工具,但找不到任何东西。我还研究了Android TensorFlow-lite库是否可以导入这些图形模型,但是我也没有找到任何东西。

我尝试使用以下找到的here的Python脚本对其进行转换,但它需要一个.pb文件:

import tensorflow as tf

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

1 个答案:

答案 0 :(得分:1)

TFLite转换器当前不支持加载tf-js模型。但是,您可以尝试使用tflite自己的posenet模型(https://www.tensorflow.org/lite/models/pose_estimation/overview

希望这可以使事情变得更容易。