rshf

Remote sensing pretrained models easy loading using huggingface -- PyTorch (for fast benchmarking)

Installation:

pip install rshf

Example:

from rshf.satmae import SatMAE
model = SatMAE.from_pretrained("MVRL/satmae-vitlarge-fmow-pretrain-800")
input = model.transform(torch.randint(0, 256, (224, 224, 3)).float().numpy(), 224).unsqueeze(0)
print(model.forward_encoder(input, mask_ratio=0.0)[0].shape)

TODO:

• Add transforms for each model
• Add Documentation (https://rshf-docs.readthedocs.io/en/latest/)
• Add initial set of models

Citations

List of models available here: Link