@Dwight_Foster Hi, I know it’s been some time since this post has been active.
But I tried your method and I have some doubts:


model.roi_heads.box_predictor.cls_score = nn.Linear(1024,len(coco_names). Here we are just telling our model to predict for 3 classes but how does the model know that the classes should be ‘background’, ‘person’ and a ‘dog’?