RS pretrained models in huggingface style
Project description
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)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
rshf-0.0.1.tar.gz
(15.8 kB
view hashes)
Built Distribution
rshf-0.0.1-py3-none-any.whl
(22.9 kB
view hashes)