通过TFLiteConverter将tensor_forest模型转换为tflite模型?

时间:2019-07-17 08:44:21

标签: tensorflow machine-learning random-forest tensorflow-lite

如何使用tf.lite.TFLiteConverter从可能在Android智能手机中使用的tensor_forest模型中获取tflite。

我发现contrib目录(link)中的tensor_forest。这里有一个使用此模型的示例(link)。

我添加以下代码以保存模型:

saver = tf.train.Saver(save_relative_paths=True, max_to_keep=10)
checkpoint_prefix = 'checkpoints/model'

for i in range(1, num_steps + 1):
  (skip)
  if i % 50 == 0 or i == 1:
    saver.save(sess, checkpoint_prefix, global_step=i)
  (skip)

它在检查点目录中生成了“检查点”,“ model-2000.data-00000-of-00001”,“ model-2000.index”,“ model-2000.meta”。但是我无法通过引用tf.lite.TFLiteConverter的使用来转换它,因为它似乎需要pb文件或缺少一些参数。

如何以正确的方式生成tflite?

0 个答案:

没有答案