DEIM icon indicating copy to clipboard operation
DEIM copied to clipboard
Published 3 months ago • verified publisher icon Intellindust-AI-Lab

One class custom dataset

Open Jobiis opened this issue 1 year ago • 6 comments

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

Jobiis avatar Dec 25 '24 19:12 Jobiis

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

an additional class might be background

faicaiwawa avatar Dec 26 '24 17:12 faicaiwawa

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

an additional class might be background

So you would suggest when dataset has 1 class to set num_classes to 2?

Jobiis avatar Dec 26 '24 17:12 Jobiis

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

an additional class might be background

So you would suggest when dataset has 1 class to set num_classes to 2?

yes

faicaiwawa avatar Dec 26 '24 17:12 faicaiwawa

@ShihuaHuang95 could you confirm?

Jobiis avatar Dec 26 '24 17:12 Jobiis

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

me too

Bigtuo avatar Jan 09 '25 02:01 Bigtuo

same issue, can someone give the explaination? THANKS!

Vanxy999 avatar Jan 14 '25 03:01 Vanxy999

Hello, I am using a dataset that has only one class. I have modified the .yml file and set num_classes to 1, but I get errors. When I set though num_classes to 2 the training begins as expected. Can you provide some info on what to do with datasets that have 1 class?

The bug still exists, and I hope the authors can fix it soon. The robustness of academic code is still not as reliable as that of commercial closed-source code.

IzouGend avatar Dec 02 '25 09:12 IzouGend