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 supports models from the Prithvi, TerraMind, and Granite series as well as models from TorchGeo and timm.

This library provides

Ease-of-use and all the functionality from Lightning and TorchGeo.
A modular model factory that combines any backbone with different decoders for full flexibility.
Ready-to-use tasks for image segmentation, pixelwise regression, classification, and more.
Multiple abstraction levels and inference pipelines to power enterprise applications.