保存经过训练的模型后缺少 pbtxt

Question

我想做的是将训练有素的 CNN 转换为 TfLite，并在我的 android 应用中使用它。据我所知，我需要 .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)

获取文字版。