search

szaza / android-yolo-v2

Android YOLO real time object detection sample application with Tensorflow mobile.
Do What The F*ck You Want To Public License
227 stars 86 forks source link

Node Output does not exist in model #31

Open Vivek-Thakare opened 4 years ago

Vivek-Thakare commented 4 years ago

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 4 years ago

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

Vivek-Thakare commented 4 years ago

Hello!.Thank you for the response. Actually we trained dataset containing four object using tensorflow 1.2...the output of that was .pb file,there was no labels.txt file...could the error be because of that ?

On Sat, Feb 1, 2020, 12:11 AM Zoltán Szabó notifications@github.com wrote:

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

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/szaza/android-yolo-v2/issues/31?email_source=notifications&email_token=AKKNZ5PF6GM5M27YIR7ZO3DRARWHNA5CNFSM4KEJW26KYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEKPTHSA#issuecomment-580858824, or unsubscribe https://github.com/notifications/unsubscribe-auth/AKKNZ5JQUOEVFNUSKYH3IV3RARWHNANCNFSM4KEJW26A .

szaza commented 4 years ago

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

Vivek-Thakare commented 4 years ago

Hello!, I created the label file...after i build it gives the same "node output does not exitst" but when i paste your pb file it works...and when i use my file it throws error.

On Sat, Feb 1, 2020, 3:34 PM Zoltán Szabó notifications@github.com wrote:

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

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/szaza/android-yolo-v2/issues/31?email_source=notifications&email_token=AKKNZ5KSLL5AM6PY7XXSGVDRAVCMPA5CNFSM4KEJW26KYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEKQZG2I#issuecomment-581014377, or unsubscribe https://github.com/notifications/unsubscribe-auth/AKKNZ5PPRVUOINBNE2XXR7LRAVCMPANCNFSM4KEJW26A .