How to get Class label from Mosaic augmentation in Object Detection Dataloader?
问题 NOTE: I couldn't think any better Title name, you're most welcome to edit or suggestion. Update Direct Colab Link. Just grab the given dummy data set and load it to colab. I'm trying to train an object detection model for a multi-class problem. In my training, I am using the Mosaic augmentation, Paper, for this task. In my training mechanism, I'm a bit stuck to properly retrieve the class labels of each category, as the augmentation mechanism randomly picks the sub-portion of a sample.