OlmoEarth#

torchgeo.models.olmoearth_v1(weights=None, **kwargs)[source]#

OlmoEarth v1 model.

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

This model requires the following additional library to be installed:

Added in version 0.10.

Parameters:

  • weights (OlmoEarthV1_Weights | None) – Pre-trained weights. If None, model is randomly initialized.

  • **kwargs (Any) – Passed to olmoearth_pretrain_minimal.OlmoEarthPretrain_v1 (e.g. model_size, max_patch_size).

Returns:

An OlmoEarth v1 model.

Return type:

Module

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

Bases: WeightsEnum

OlmoEarth v1 pre-trained weights.

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

Added in version 0.10.

__new__(value)[source]#