Srikumar26 commited on
Commit
a1a4699
1 Parent(s): 792e911

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+ Model: SatMAE++ (https://arxiv.org/abs/2403.05419)
5
+
6
+ Variant: ViT-L_pretrain_fmow_rgb
7
+
8
+ Example Usage:
9
+ ```python
10
+ from huggingface_hub import hf_hub_download
11
+ import torch
12
+
13
+ hf_hub_download("MVRL/satmaepp_ViT-L_pretrain_fmow_rgb", "model.py", local_dir=".")
14
+
15
+ from model import MaskedAutoencoderViT
16
+
17
+ model = MaskedAutoencoderViT.from_pretrained("MVRL/satmaepp_ViT-L_pretrain_fmow_rgb")
18
+
19
+ print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shape)
20
+ ```