CODEWITHSUNDEEP
pythontensorflowtensorflow-slim
V.Hunon
V.Hunon

Reputation: 320

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

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.

Upvotes: 0

Views: 675

Answers (1)

Guinther Kovalski
Guinther Kovalski

Reputation: 1919

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

Upvotes: 1

Related Questions