pierluigiferrari / ssd_keras

A Keras port of Single Shot MultiBox Detector
Apache License 2.0
1.86k stars 938 forks source link

Pascal VOC Training Person Detection #388

Open aidansmyth95 opened 2 years ago

aidansmyth95 commented 2 years ago

How do I manipulate the data generator in the training SSD300 script to train only on background and person class? When I change n_classes = 1 and include_classes=[0,15] I am met by an error:

IndexError: index 15 is out of bounds for axis 0 with size 2