Additional categories in imagenet

I did dog breed lesson. I know that we can finetune the last layer of the model (resnet50) to recognize dog breed, but this new model will not recognize its previous classes anymore like bird, cat, etc.

Now lets say I want resnet50 to recognize all resnet50 classes including an extra categories that I trained. How do I do that without retraining all imagenet photos? Example code reference would be awesome.

Thanks

I do not think there is an easy way of doing this like you describe.

The simplest way to achieve this would be to combine your custom dataset with imagenet and train a pretrained model on combined data.

1 Like