Train and predict multiple classes detecting for single object

THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection

https://arxiv.org/abs/2405.14458

GNU Affero General Public License v3.0

9.34k stars 874 forks source link

Train and predict multiple classes detecting for single object #116

Open SutirthaChakraborty opened 3 months ago

SutirthaChakraborty commented 3 months ago

How can we train a model to detect multiple class of a single object. For example : an image of bee, it should be able to detect bee and 'insect' an image of rat, it should be able to detect 'rat' and 'rodent'

What should be the training code and what should be the python code for inference for this kind of multi-class where each class will be independent of each other.

leonnil commented 3 months ago

Thank you for your interest! Hope this https://github.com/ultralytics/ultralytics/issues/489 would help.