Support additional SatlasPretrain models. about torchgeo HOT 6 OPEN

Okay, sounds good. I will just start by adding the additional backbones in a PR, and if you all come up with a plan or skeleton for decoders, I can make a PR for that later.

And yes, our data and model checkpoints are under ODC-BY and the code is Apache 2.0.
Thank you!

from torchgeo.

@piperwolters I added the previous Satlas checkpoints. If you need any help let me know! We are also active on the TorchGeo slack channel if you want to discuss.

from torchgeo.

@adamjstewart That first PR only added single-image Swin backbones. Would like to add multi-image backbones and decoders eventually, but don't necessarily need to keep this issue open for that.

from torchgeo.

Yes, we would love to get the rest of the SatlasPretrain models in TorchGeo so they're as easy as possible for our users to access!

My preference would be for option 1. This avoids adding "yet another dependency" and integrates nicely with our Lightning trainers. Note that we've really only added backbones in TorchGeo so far. We would like to add decoders as well, but haven't yet put a lot of thought into how the API should work.

P.S. When we added the first couple SatlasPretrain models, we assumed they were under the Apache-2.0 license, but based on this new repo it seems they are actually ODC-BY. Is this correct? I'll update our documentation if so.

FYI @favyen2

from torchgeo.

Is this completed by #1884?

from torchgeo.

Nah, let's keep it open. You could add a bullet point list to the issue like:

• Swin Sentinel single-image
• Swin Sentinel multi-image

if you want to track progress on completion.

from torchgeo.