How to find the class labels of the detected objects (img.prediction.labels) only returns the numeric value

[maninka123]

Is your feature request related to a problem? Please describe.

A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]

Describe the solution you'd like

A clear and concise description of what you want to happen.

Describe alternatives you've considered

A clear and concise description of any alternative solutions or features you've considered.

Additional context

Add any other context or screenshots about the feature request here.

[dagshub]

Join the discussion on DagsHub!

[Louis-Dupont]

Each image prediction also includes an attribute class_names with all the classes names that the model tries to predict, here is an example:

images_predictions = model.predict(IMAGES)

for image_prediction in images_predictions:
    class_names = image_prediction.class_names  # List of class names the model tries to predict for in the image
    labels = image_prediction.prediction.labels
    confidence = image_prediction.prediction.confidence
    bboxes = image_prediction.prediction.bboxes_xyxy
    for i, (label, conf, bbox) in enumerate(zip(labels, confidence, bboxes)):
        print("prediction: ", i)
        print("label_id: ", label)
        print("label_name: ", class_names[int(label)])
        print("confidence: ", conf)
        print("bbox: ", bbox)
        print("--" * 10)

[Louis-Dupont]

@maninka123 we just released a new documentation on model.predict() was added, check it out for more information, and feel free to let us know if anything is unclear.