Что я пытаюсь сделать, так это преобразовать мою обученную 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
файлы. Нет пбтхт.
Есть ли способ преобразовать обученную сеть в tflite без pbtxt или я могу получить pbtxt из этих файлов?
Спасибо
Просто выполните:
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)
чтобы получить текстовую версию.