python-3.x - 保存经过训练的模型后缺少 pbtxt

Question

我想要做的是将我训练有素的 CNN 转换为 TfLite 并在我的 android 应用程序中使用它。AFAIK 我需要 .pbtxt 来冻结参数并进行转换。

但是，当我使用此标准代码保存网络时：

saver = tf.train.Saver(max_to_keep=4)
saver.save(sess=session, save_path="some_path", global_step=step)

我只得到

.data 
.index 
.meta
checkpoint

文件。没有pbtxt。

有没有办法在没有 pbtxt 的情况下将经过训练的网络转换为 tflite，或者我可以从这些文件中获取 pbtxt 吗？

谢谢

Answer 1

只需执行：

tf.train.write_graph(session.graph.as_graph_def(),
                                     "path",
                                     'model.pb',
                                     as_text=False)

获取 .pb 或

tf.train.write_graph(session.graph.as_graph_def(),
                                     "path",
                                     'model.pbtxt',
                                     as_text=True)

获取文本版本。