Labels for models? #527
Unanswered
tomCarnevale
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi, apologies if this is not the right place for this question
I am trying to learn how to use the ONNX models, specifically in Unity Barracuda and I am new to Machine Learning in general. I have managed to use some of the models present in this repository, but am unsure where I can find labels for the models.
For example, I am using
resnet50-v1-12
to classify a single image and I can get a tensor from the output with TensorShape dimensions of batch 1, width 1, height 1, and 1000 channels. The channels representing the score, if I'm reading that right, but I have no way of mapping those scores to any human readable data, like labels. In this case, item 285 has the highest score. if I had a label text file, I could look up the meaning of the 285th element. Is it cat? (I hope so)Is there any label data present in an onnx? If not, how can I track down a list of labels for a given model?
I am sort of assuming I may need to find the original and convert it to onnx myself.
Beta Was this translation helpful? Give feedback.
All reactions