search

weinman / cnn_lstm_ctc_ocr

Tensorflow-based CNN+LSTM trained with CTC-loss for OCR
GNU General Public License v3.0
498 stars 170 forks source link

What is the output_node_name in the model? #37

Closed Albertcc closed 5 years ago

Albertcc commented 5 years ago

I want to save the model in PB file mode,Then I can call it in Vc++,But I can't find the output_node_name. How could I found it?

weinman commented 5 years ago

The name given is logits but if you want the complete "path" to the name you'll have to search for it.

Albertcc commented 5 years ago

Also, I think the output is logits. But then the name "rnn/logits" is a subgraph. I tried the name "rnn/logits/Relu" as the output_node_name. Try to copy the tensor "logits" in train as the output ...... Result = tf.identity(logits,name="output") ...... And all the above, I can't get the same right result as restored the model in CPKT mode.

What should I do??? Help me.

weinman commented 5 years ago

I cannot provide a Tensorflow how-to. You might try asking on StackOverflow.

shodow commented 5 years ago

@Albertcc I have the same problem,have you solved it?

Albertcc commented 5 years ago

No. When I debug the code, I can see the output tensor named "rnn/logits/Relu:0", But still , I can't get the right output.

kapsmeh commented 4 years ago

@Albertcc did you find the input and output nodes yet??