search

datvuthanh / HybridNets

HybridNets: End-to-End Perception Network
MIT License
585 stars 120 forks source link

MultiLabel Classification #15

Open abhigoku10 opened 2 years ago

abhigoku10 commented 2 years ago

@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 commented 2 years ago

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 commented 2 years ago

I'll close this topic. Thank you!