Small and easily modifiable notebook to extract embeddings from pre trained resnet50.
The notebook extract embeddings for CIFAR100
, using a pretrained resnet50.
For the pretrained model we use https://github.com/facebookresearch/swav , thus the embeddings are learned unsupervised on imagenet. You can easily swap it for other models of your choosing.
If you do use SWAV
, make sure you cite them properly.