Aurora

torchgeo.models.aurora_swin_unet(weights=None, *args, **kwargs)

Aurora model.

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

• https://arxiv.org/abs/2405.13063

This dataset requires the following additional library to be installed:

• microsoft-aurora to load the models.

Added in version 0.8.

Parameters:

• weights (Aurora_Weights | None) – Pre-trained model weights to use.

• *args (Any) – Additional arguments to pass to aurora.Aurora

• **kwargs (Any) – Additional keyword arguments to pass to aurora.Aurora

Returns:

An Aurora model.

Return type:

Module

class torchgeo.models.Aurora_Weights(*values)

Bases: WeightsEnum

Aurora weights.

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

• https://arxiv.org/abs/2405.13063

Added in version 0.8.

__new__(value)