Github
Huggingface
文档TerraTorch
:book: 文档
概述
TerraTorch 是一个基于 PyTorch Lightning 和地理空间数据领域库 TorchGeo 的库。
TerraTorch 的主要目的是为地理空间基础模型提供一个灵活的微调框架，可以在不同抽象层次上进行交互。
该库提供：
- 轻松访问开源预训练的地理空间基础模型骨干网络（如 Prithvi、SatMAE 和 ScaleMAE），以及 timm（Pytorch 图像模型）或 SMP（带有预训练骨干网络的 Pytorch 分割模型）包中可用的其他骨干网络，还有像 granite-geospatial-biomass 这样的微调模型
- 用于图像分割、分类和像素级回归微调任务的灵活训练器
- 通过灵活的配置文件启动微调任务
安装
Pip
为了使用 pyproject.toml 文件，需要确保 pip>=21.8。如有必要，使用 python -m pip install --upgrade pip 升级 pip。
对于稳定的版本，使用 pip install terratorch。
如果你想获取主分支的最新版本，可以使用 pip install git+https://github.com/IBM/terratorch.git 安装库。
另一种方法是使用 pipx 通过 pipx install terratorch 进行安装，这会创建一个隔离环境，允许用户将应用程序作为常用 CLI 工具运行，无需安装依赖项或激活环境。
TerraTorch 需要安装 gdal，这可能是一个相当复杂的过程。如果你的系统上没有设置 GDAL，我们建议使用 conda 环境，并通过 conda install -c conda-forge gdal 安装它。
要作为开发者安装（例如扩展库），克隆此仓库，使用 pip install -r requirements/required.txt -r requirements/dev.txt 安装依赖项，然后运行 pip install -e .
快速入门
要开始使用，请查看快速入门指南
开发者指南
查看架构概览
