Hi @JamesDickens ,
Thank you for your answer. I do not want to train the model with a new dataset. I want just that model detects my wishes classes which are available in the coco dataset:
coco_names = ['__background__', 'person', 'dog']

and ignores other classes.
I am tried to set the num_classes to 3 by loading the model it did not work.