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
-
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