SSD-on-Custom-Dataset icon indicating copy to clipboard operation
SSD-on-Custom-Dataset copied to clipboard

Published 20 hours ago • verified publisher icon Coldmooon

Train For negative images

Open SikhaS opened this issue 6 years ago • 2 comments

I want to train mobileNet SSD on my own dataset for a person class only. And i wanted to know about how should i train it for negative images that i dont wish to get detected?

Any help will be appreciated.

SikhaS avatar May 10 '19 06:05 SikhaS

@SikhaS SSD has considered the class of background. If you have only one class, just set num_classes in ./examples/ssd/ssd_xxx.py to 2.

Coldmooon avatar May 11 '19 10:05 Coldmooon

@Coldmooon Thank you for the solution. However can you explain a bit more about num_Classes = 2 and what will be the outcome of changing this to 2?

SikhaS avatar May 13 '19 10:05 SikhaS