我有导出器,它生成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))