get_unique_detection_classes misses a detection class

[grzegorzj]

Search before asking

• I have searched the Multimodal Maestro issues and found no similar bug report.

Bug

I ran the example notebook with my own dataset hosted on Roboflow (https://app.roboflow.com/12x2experiments/rectangles-test-2):

https://colab.research.google.com/drive/1kO_oZhzfPFmuJqoH4ov8oeGYGfSaHAu9?usp=sharing

and ran into an issue in which detection class that was detected in the image is missing from overall possible detection classes:

class_name = 'rounded_rectangle'
classes = ['rectangle', 'triangle']

ValueError: 'rounded_rectangle' is not in list

It seems like somewhere in the chain, get_unique_detection_classes fails to correctly return all three classes in the dataset.

Environment

• Google Collab, NVidia A100

Minimal Reproducible Example

(attached collab notebook)

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!