如何使用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?