获取冻结图的输出(frozen_model.pb)

时间:2018-09-06 13:18:52

标签: tensorflow tensorflow-serving nvidia-digits

我正在使用带有数字的张量流。我想扩展服务功能的位数。

我已经可以通过训练在运行时为模型提供服务了,但要点是,我想在模型已经训练之后实现这种可能性。为此,我需要会话,输入和输出变量。

我正在从检查点文件中获取

会话和输入。只是我有问题要获取输出。我知道输出会以常量形式存储在冻结图中。

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。

谢谢,谢谢。

0 个答案:

没有答案