search

ahmetozlu / tensorflow_object_counting_api

🚀 The TensorFlow Object Counting API is an open source framework built on top of TensorFlow and Keras that makes it easy to develop object counting systems!
https://www.youtube.com/watch?v=yT_1eKJTdfk
MIT License
1.32k stars 545 forks source link

Index out of Range Error After Reducing number of class_names #87

Closed Rahul214 closed 3 years ago

Rahul214 commented 3 years ago

Hi @ahmetozlu, we are getting list index out of range after reducing no. of class_names while running single_image_object_counting.py file in mask_rcnn_counting_api_keras_tensorflow folder. Can you please help us? Thanks! error class

ahmetozlu commented 3 years ago

Since you are using coco trained model, there have to be 80 classes in default. If you need to update classes, you have to perform transfer learning with your own data for your custom classes. Then, you should inference the model and use it. You also need to update configurations such as NUM_CLASSES, since you will use your own custom trained model..