2016-02-06 4 views
8

Начальная сеть в Android Tensorflow Demo хранит модель в виде протобуфа (tensorflow_inception_graph.pb). Я хотел бы заменить эту сеть на другую.Преобразование обучаемой модели Tensorflow в protobuf

Есть ли документация или пример где-то о том, как преобразовать обученную сеть TensorFlow в Python в .pb? Я знаю TensorFlow's Saver, но это, похоже, используется для сохранения промежуточного состояния тренировки. Не знаете, как это работает, если модель уже обучена.

+0

если один из указанных ответов верен или вы нашли другое решение, и обновление здесь было бы неплохо – BraveHeart

ответ

4

tf.Graph.as_graph_def() извлекает сериализованную граф. Затем вы просто записываете его в файл.

Смежные вопросы