Skip to content

Small and easily modifiable notebook to extract embeddings from pre trained resnet50

Notifications You must be signed in to change notification settings

dinarior/resnet_embeddings_extractor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Resnet Embedding Extractor

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.

About

Small and easily modifiable notebook to extract embeddings from pre trained resnet50

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published