Scale-MAE#

torchgeo.models.ScaleMAE(res=1.0, *args, **kwargs)[source]#

Custom Vision Transformer for Scale-MAE with GSD positional embeddings.

This is a ViT encoder only model of the Scale-MAE architecture with GSD positional embeddings.

If you use this model in your research, please cite the following paper:

class torchgeo.models.ScaleMAELarge16_Weights(*values)[source]#

Bases: WeightsEnum

Scale-MAE Large patch size 16 weights.

Added in version 0.6.

__new__(value)#