Checkpoint

• Single-Image Swin Backbones
• Multi-Image Swin Backbones
• Other Backbones (Swin-Tiny, Resnets)
• Decoders

Summary

I see that a couple SatlasPretrain models have been merged and will be included in the 0.6.0 release, which is great!
I'm hoping that additional pretrained weights and models could be integrated, specifically the Sentinel-1 and Landsat models as well as models with varying number of input images and bands.

We recently released the satlaspretrain-models package which makes it easy to load the various pretrained models. I could make a PR to torchgeo to include more pretrained models, either 1) using the code that was put in place in the above PR or 2) using this new satlaspretrain-models package. Wondering if you all have a preference?
Thanks!

Rationale

SatlasPretrain models improve performance and speed up training on downstream tasks, as opposed to training from scratch. It would be great to have these models easily accessible for torchgeo users.

Implementation

Either follow the pattern from this merged PR or utilize the satlaspretrain-models package.

Alternatives

No response

Additional information

No response