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
.
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
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.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | cae627761b9ead6ddd3e3759b4864a68ff30e026a883533550a1f30fdfb00ec5 |
|
MD5 | 98404bd13b4f2bfd751f2f20c529a2e8 |
|
BLAKE2b-256 | c250f080272f9b28da25dec1c1ca8e0cde4fd25df2c05b09f475110bec4ced1a |