Welcome to TerraTorch

Overview

The purpose of this library is twofold:

To integrate prithvi backbones into the TorchGeo framework.
To provide generic LightningDataModules that can be built at runtime.
To build a flexible fine-tuning framework based on TorchGeo which can be interacted with at different abstraction levels.

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, and to a certain extent TorchGeo

To get started, check out the quick start guide