Srikumar26's picture
Create README.md
a1a4699 verified
|
raw
history blame
No virus
495 Bytes
---
license: apache-2.0
---
Model: SatMAE++ (https://arxiv.org/abs/2403.05419)
Variant: ViT-L_pretrain_fmow_rgb
Example Usage:
```python
from huggingface_hub import hf_hub_download
import torch
hf_hub_download("MVRL/satmaepp_ViT-L_pretrain_fmow_rgb", "model.py", local_dir=".")
from model import MaskedAutoencoderViT
model = MaskedAutoencoderViT.from_pretrained("MVRL/satmaepp_ViT-L_pretrain_fmow_rgb")
print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shape)
```