microsoft
/

rad-dino

Image Feature Extraction

Model card Files Files and versions

fepegar commited on Oct 2

Commit

5dc5f2b

·

verified ·

1 Parent(s): 49d20bf

Fix variable names

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ pip install rad-dino
 >>> image = download_sample_image()
 >>> image
 <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=2765x2505 at 0x7CCD5C014050>
->>> cls_token, patch_tokens = encoder.extract_features(image)
 >>> cls_embeddings.shape, patch_embeddings.shape
 (torch.Size([1, 768]), torch.Size([1, 768, 37, 37]))
 ```

 >>> image = download_sample_image()
 >>> image
 <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=2765x2505 at 0x7CCD5C014050>
+>>> cls_embeddings, patch_embeddings = encoder.extract_features(image)
 >>> cls_embeddings.shape, patch_embeddings.shape
 (torch.Size([1, 768]), torch.Size([1, 768, 37, 37]))
 ```