szaza/android-yolo-v2

Node Output does not exist in model

Vivek-Thakare opened this issue · 4 comments

This is my custom pb file for detection for single object.I'm getting this error.
Process: org.tensorflow.yolo, PID: 23066
java.lang.RuntimeException: Node 'output' does not exist in model 'file:///android_asset/tiny-yolo-voc-graph.pb'
at org.tensorflow.contrib.android.TensorFlowInferenceInterface.graphOperation(TensorFlowInferenceInterface.java:257)

szaza commented

Hi, it seems that the path is invalid. Could you crosscheck it?

szaza commented

Tensorflow doesn't generate the labels.txt file automatically, but you can do it manually, you only have to enumerate your classes and place the file into the assets directory.
Example: https://github.com/szaza/android-yolo-v2/blob/master/assets/tiny-yolo-voc-labels.txt