Replies: 1 comment
-
@pasmai ONNX models don't have class names attached as metadata, but you can insert your own class names in DetectMultiBackend(): Line 294 in 562191f |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hello everyone
I would like that when predicting a
.onnx
model that the labels are also directly shown in the image.python detect.py --weights /models/trained/yolov5m.pt
generates an image that contains labels.python detect.py --weights /models/trained/yolov5m.onnx
generates an image that where the boxes are labelled asclass0
Is there a good way to re-use or include the labels from the
.yaml
datasets?Beta Was this translation helpful? Give feedback.
All reactions