What are the changes needed to use more than 5 classes?

Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT

188 stars 91 forks source link

What are the changes needed to use more than 5 classes? #33

Open DongNaeSwellfish opened 5 years ago

DongNaeSwellfish commented 5 years ago

https://github.com/Ghustwb/MobileNet-SSD-TensorRT/issues/9 This issue seems like it only useful if the number of classes is less than 5. I want to use the full 21 classes, so I applied app the changes on the link, but they didn't help. Should I make a new caffemodel.1.tensorcache? If I should, how can I make tensorcache?