保存训练有素的模型后，缺少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)

获取文本版本。