TerraTorch - A model training toolkit for geospatial tasks
Project description
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
.
For a stable point-release, use pip install terratorch
.
If you prefer to get the most recent version of the main branch, 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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for terratorch-0.99.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a18b51ba256cf365f14f40e0800ee829cc47e62b98ff011b147d53531e4392fe |
|
MD5 | 3fadff7c7729e1954835bc136091efda |
|
BLAKE2b-256 | 8fc71676d175a190802aaf91af07ab8623bc1ffb5b4e5e3b009ff0cad24db376 |