Welcome to TerraTorch#

Overview#

The purpose of this package is to build a flexible fine-tuning framework for Geospatial Foundation Models (GFMs) based on TorchGeo and Lightning which can be employed at different abstraction levels. It currently supports models from the Prithvi and Granite series, and also have been tested with others models available on HuggingFace.

This library provides:

• All the functionality in TorchGeo.
• Easy access to Prithvi, timm and smp backbones.

• Flexible trainers for Image Segmentation, Pixel Wise Regression and Classification (more in progress).

  • Launching of fine-tuning tasks through powerful configuration files.

  A good starting place is familiarization with PyTorch Lightning, which this project is built on. TorchGeo is also an important complementary reference.

  Check out the architecture overview for a general description about how TerraTorch is organized.

Quick start#

To get started, check out the quick start guide

License#

TerraTorch is distributed under the terms of License Apache 2.0, see here for more details.