VOC with Fewer Classes

Open bspivey opened this issue 4 years ago • 1 comments

How can we train on fewer classes with the VOC dataset? I have changed the VOC_CLASSES dictionary to two labels and updated cfg['num_classes'] to 3, yet I still see the algorithm looking for other labels.

Sep 27 '21 02:09 bspivey

I think in this case you have to either adjust the training labels or generate new training labels with only the desired classes

Jan 03 '22 10:01 ksmdnl