rshf

RS pretrained models in huggingface style


Keywords
Remote, Sensing, Huggingface
License
Apache-2.0
Install
pip install rshf==0.0.5

Documentation

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")
print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shape)

List of models available here: Link