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)
List of models available here: Link
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.5.tar.gz
(2.0 MB
view hashes)
Built Distribution
rshf-0.0.5-py3-none-any.whl
(2.0 MB
view hashes)
