One class custom dataset

[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?

[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

[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?

[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?

yes

[Jobiis]

@ShihuaHuang95 could you confirm?