Tensorflow将训练好的模型从save_model.pb转换为model.ckpt格式

时间:2020-04-20 02:52:01

标签: python tensorflow object-detection

我有导出器,它生成model.ckpt *文件。后来,我还使用tf.estimator.BestExporter向管道添加了一个最佳的模型保护程序。

因此,较早的出口商吐出以下文件

model.ckpt-56846.data-00000-of-00001
model.ckpt-56846.index
model.ckpt-56846.meta

新的最佳模型导出器将吐出以下文件:

saved_model.pb
variables.data-00000-of-00001
variables.index

我的管道中的下一步是运行一次eval,然后导出推理图。现在如何将saved_model.pb转换为model.ckpt格式,以便我可以继续使用python object_detection/model_main.py进行评估和使用object_detection/export_inference_graph.py导出冻结的推理图。

我担心的是Tensorflow 1.10中的以下功能可能不适用于saved_model.pb

estimator.evaluate(input_fn, 
                   None, 
                   checkpoint_path=tf.train.latest_checkpoint(FLAGS.checkpoint_dir))

0 个答案:

没有答案
相关问题