birder-project
/

vit_l16_mim

Image Feature Extraction

Model card Files Files and versions Community

hassonofer commited on Feb 25

Commit

1430392

·

verified ·

1 Parent(s): 9c9b813

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ import torch
 import birder
 from PIL import Image
-(net, _, signature, rgb_stats) = birder.load_pretrained_model("vit_l16_mim_200", inference=True)
 # Get the image size the model was trained on
 size = birder.get_size_from_signature(signature)

 import birder
 from PIL import Image
+(net, _, signature, rgb_stats) = birder.load_pretrained_model("vit_l16_mim_400", inference=True)
 # Get the image size the model was trained on
 size = birder.get_size_from_signature(signature)