MultiLabel Classification

[abhigoku10]

@datvuthanh @xoiga123 Thanks for sharing the code base, it is really helpful , but i had a few queries

1. Can we have multi-label classification also from the existing code? as I see two things one is that in the annotations there is attribute:{"trafficator": green} and in the loss.py there is a macro MULTILABEL mode is there
2. I am looking for a single bounding box with multilabel classification output so what are the modifications to be made in the existing code

Thanks in advance

[datvuthanh]

Hi @abhigoku10, you just change number of classes in bdd100k.yaml. Also, you need to modify dataset.py to load your dataset. I don't think it is so difficult.

[datvuthanh]

I'll close this topic. Thank you!