Is there a way to train only 1 class?

maikherbig / AIDeveloper

GUI-based software for training, evaluating and applying deep neural nets for image classification

BSD 2-Clause "Simplified" License

111 stars 20 forks source link

Is there a way to train only 1 class? #60

Open CptJackieSparrow opened 2 months ago

CptJackieSparrow commented 2 months ago

I'd like to know if it's possible to train only 1 class? I'm having "Need at least two classes to fit. Please specify .rtdc files and corresponding indices" error when i try to train one. Ty again for making the public!

maikherbig commented 2 months ago

It would be possible to run such a training, but it would probably immediately overfit to always return "0" (class number 0), independent of the input image. Hence, it is practically useless. During training of a classification model, the algorithm learn to spot the differences between classes of objects.