Получить вывод из замороженного графика (frozen_model.pb) - PullRequest
Новая
       62

Получить вывод из замороженного графика (frozen_model.pb)

0 голосов
/ 06 сентября 2018

Я использую тензорный поток с цифрами. Я хотел бы расширить цифры для обслуживания функциональности.

Я уже могу обслуживать модель во время выполнения с обучением, но суть в том, что я хотел бы сделать эту возможность после того, как модель уже обучена. Для этого мне нужны переменные сессии, ввода и вывода.

Сессия и ввод Я беру из файлов контрольных точек. Просто у меня есть проблема, чтобы получить вывод. Я знаю, что выходные данные хранятся в замороженном графике как константы. 

freeze_graph.freeze_graph(
            input_graph=graphdef_path,
            input_saver='',
            input_binary=True,
            input_checkpoint=checkpoint_path,
            **output_node_names=out_name**,
            restore_op_name="save/restore_all",
            filename_tensor_name="save/Const:0",
            output_graph=path_frozen,
            clear_devices=True,
            initializer_nodes="",
        )

Но я не знаю, как их получить и возможно ли создать build_tensor_info из констант.

Буду признателен за помощь, спасибо.

...