TerraTorch：地空间基础模型工具包
计算机视觉与模式识别
2025-03-27 v1 机器学习
摘要
TerraTorch 是一个基于 PyTorch Lightning 构建的遥感、气象和气候数据细化与基准测试工具包。它集成了领域特定数据模块、预定义任务以及模块化模型工厂，后者可将任意 backbone 与多样化的解码器头配对。这些组件允许研究人员和实践者通过仅编辑训练配置即可进行无代码细化。通过整合最佳实践并内置自动超参数优化扩展 Iterate，TerraTorch 减少了在新地球观测用例中细化或基准测试模型所需的专业知识和时间。此外，TerraTorch 直接集成 GEO-Bench，可实现对地空间基础模型的系统性、可重复的基准测试。TerraTorch 采用 Apache 2.0 许可证开源，地址为 https://github.com/IBM/terratorch，可通过 pip install terratorch 安装。
引用
@article{arxiv.2503.20563,
title = {TerraTorch: The Geospatial Foundation Models Toolkit},
author = {Carlos Gomes and Benedikt Blumenstiel and Joao Lucas de Sousa Almeida and Pedro Henrique de Oliveira and Paolo Fraccaro and Francesc Marti Escofet and Daniela Szwarcman and Naomi Simumba and Romeo Kienzler and Bianca Zadrozny},
journal= {arXiv preprint arXiv:2503.20563},
year = {2025}
}
备注
IGARSS 2025