Multiple Classes

[VegetableDogKing]

Hi,
I want to ask that how can we train for multiple classes, do we just pass in two classes at the same time, or we train for one classes first and use the weigth to train for another.

Thank you

[phsilvarepo]

Hello there,
What you have to do is train the model for each class you want to detect, then in the inference add the multiple weights you trained for each class.

Good luck

[TontonTremblay]

@phsilvarepo is correct, you could modify the outputs to do 2 at the same time, I have done it in the past, but not worth the time you are going to spent on it wont be worth it. Easier to train one at the time :D