2ee326e21cd8559c077cef408d32355d56484bb1,examples/convert.py,,export_cnn,#,14

Before Change

    output = "cnn.pb"
    graph_io.write_graph(frozen, ".", output, as_text=False)
    print("saved the frozen graph (ready for inference) at: ", output)


class PredictionInputProvider(tfe.io.InputProvider):
    def provide_input(self) -> tf.Tensor:

After Change

    pred_node_names = ["output"]
    tf.identity(x, name=pred_node_names[0])

    with tf.Session() as sess:
        constant_graph = graph_util.convert_variables_to_constants(sess,
                                                                   sess.graph.as_graph_def(),
                                                                   pred_node_names)

Instances