How do I fine tune a model on a dataset with different classes?

Question

In this paper the author fine tuned a COCO-pretrained model to PoseTrack Dataset. I want to do something similiar but I am not sure how to do that, since the number of classes (keyjoints) are not the same.

I thought about "disabling" certain classes but I am unsure how.

Answer 1

You will need to change the last layer of the nn with the number of classes of your problem, and re train for your dataset (supposing it has the same input shape). You can use the pre trained weights of the previous layer so that the convergence will be faster and you will need less data