terratorch 0.99.0

TerraTorch - A model training toolkit for geospatial tasks

TerraTorch

:book: Documentation

Overview

TerraTorch is a library based on PyTorch Lightning and the TorchGeo domain library for geospatial data. TerraTorch’s main purpose is to provide a flexible fine-tuning framework for Geospatial Foundation Models, which can be interacted with at different abstraction levels.

The library provides:

• Easy access to open source pre-trained Geospatial Foundation Model backbones (e.g., Prithvi, SatMAE and ScaleMAE and other backbones available in the timm (Pytorch image models) or SMP (Pytorch Segmentation models with pre-training backbones) packages.
• Flexible trainers for Image Segmentation, Classification and Pixel Wise Regression fine-tuning tasks
• Launching of fine-tuning tasks through flexible configuration files

Install

Pip

In order to use th file pyproject.toml it is necessary to guarantee pip>=21.8. If necessary upgrade pip using python -m pip install --upgrade pip.

Install the library with pip install git+https://github.com/IBM/terratorch.git.

TerraTorch requires gdal to be installed, which can be quite a complex process. If you don't have GDAL set up on your system, we reccomend using a conda environment and installing it with conda install -c conda-forge gdal.

To install as a developer (e.g. to extend the library) clone this repo, install dependencies using pip install -r requirements.txt and run pip install -e .

Quick start

To get started, check out the quick start guide

For developers

Check out the architecture overview