Экспорт замороженного графического файла .pb в Tensorflow 2

Question

Я пробовал пробовать альфа Tensorflow 2 и пытался заморозить и экспортировать модель в файл .pb graphdef.

В Tensorflow 1 я мог бы сделать что-то вроде этого:

# Freeze the graph.
frozen_graph_def = tf.graph_util.convert_variables_to_constants(
    sess,
    sess.graph_def,
    output_node_names)

# Save the frozen graph to .pb file.
with open('model.pb', 'wb') as f:
    f.write(frozen_graph_def.SerializeToString())

Однако это больше не представляется возможным, так как convert_variables_to_constants удален и использование сессий не рекомендуется.

Я посмотрел и обнаружил, что есть график замораживания util https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py, который работает сSavedModel экспортирует.

Есть ли еще какой-нибудь способ сделать это в Python, или я собираюсь переключить и использовать этот инструмент сейчас?

Answer 1

Я также сталкивался с этой же проблемой при переходе с tenorflow1.x на бета-версию tenenslow 2.0. Эту проблему можно решить двумя способами:

1. 1-й - это перейти к поиску документов порасположения на tenensflow2.0 и изменить синтаксис для каждой строки &
2. Чтобы использовать скрипт tf_ugrade_v2 от Google

tf_upgrade_v2 --infile your_tf1_script_file --outfile convert_tf2_file

Попробуйте выполнить приведенную выше команду, чтобы изменить свой сценарий tenorsflow1.x на tenorsflow 2.0, он решит все ваши проблемы.

Кроме того, вы можете переименовать метод (шаг вручную, ссылаясь на документацию) Переименуйте 'tf.graph_util.convert_variables_to_constants' в 'tf.compat.v1.graph_util.convert_variables_to_constants'

Проблема измерения заключается в том, что в тензорном потоке 2.0 изменилось много синтаксиса и функций, попробуйте сослаться на документы tenoflow2.0 или используйте скрипт Google tf_upgrade_v2