terratorch

一键微调遥感大模型,快速落地地理空间AI应用

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 168 🐛 待处理 61 👀 关注 25
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 878
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0 许可证,可免费使用和修改。

项目预览

terratorch 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队torchgeo
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0 许可证,可免费使用和修改。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型ai4good,ai4science,computer-vision,deep-learning,earth-observation,foundation-models,geospatial,solar-physics,weather-models
GitHub 星标★ 878
上线时间2024-05-15 00:00:00
代码更新2026-11-04
仓库体积293.1 MB
本站数据更新2026-11-10
维护状态活跃
收录时间2026-08-09
浏览次数6

项目介绍

terratorch 是一个用于微调地理空间基础模型(GFMs)的 Python 工具包。它解决了遥感领域研究人员和开发者难以高效适配大型预训练模型到特定下游任务(如土地覆盖分类、作物类型识别等)的问题。核心能力包括:提供统一的数据加载和预处理接口,支持多种主流地理空间基础模型(如 Prithvi、SatMAE 等),封装了模型微调、评估和推理的完整流程,并集成了 PyTorch Lightning 和 Hydra 等现代工具,便于配置管理和实验复现。通过简化模型接入和训练流程,terratorch 降低了地理空间 AI 的应用门槛,使非专家也能快速构建高性能的遥感应用。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 878

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:进阶 约 30 分钟 部署方式:本地 GPU 4 步

环境要求

  • Python 3.10+
  • GPU 推荐
  • GDAL 依赖

安装与启动步骤

  1. 1装包

    TerraTorch 是遥感领域的模型训练框架,基于 PyTorch Lightning。

    pip install terratorch
  2. 2要检测能力时

    带 rfdetr extra 才有目标检测模型。

    pip install terratorch[rfdetr]
  3. 3备数据

    遥感数据多为多光谱 GeoTIFF,注意波段顺序与归一化。

  4. 4跑训练

    用配置驱动的训练脚本,先在小区域上验证管线。

如何确认成功

训练能跑通并输出验证指标。

注意事项

  • Deformable-DETR 相关模型要额外编译 CUDA 算子,按文档执行 make.sh。

核心亮点

  • 统一接口支持多种主流地理空间基础模型,切换模型成本低
  • 基于 PyTorch Lightning,训练代码简洁且易于扩展
  • 提供丰富的预训练模型和示例配置,上手快

不足之处

  • 项目早期,文档和社区支持相对有限
  • 对非遥感背景的开发者,概念理解有一定门槛

适用场景

  • 土地覆盖分类与变化检测
  • 农业作物类型识别与估产
  • 环境监测与灾害评估

替代项目

TorchGeo、Raster Vision、OpenEarthMap

上一篇:Large-Time-Series-Model

下一篇:hls-foundation-os

同类项目推荐

comfyUI-Realtime-Lora 开源

在 ComfyUI 里直接训练、编辑和保存 LoRA,无需切换工具

Train and block edit and save LoRAs directly inside ComfyUI for Z-image, Flux Klein···

★ 577 2026-08-09
Dreambooth-Stable-Diffusion 开源

用几张照片,微调Stable Diffusion生成专属主体的新图像

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusio···

★ 7730 2026-08-09
Agent-FLAN 开源

让大模型学会用工具,微调数据与方法一键搞定

[ACL2024 Findings] Agent-FLAN: Designing Data and Methods of Effective Agent Tuning ···

★ 363 2026-08-09
apex-quant 开源

MoE 大模型按专家分精度压缩,省显存不掉点

Adaptive Precision for EXpert Models: MoE-aware mixed-precision quantization

★ 484 2026-10-17
OpenThoughts-Agent 开源

一套配方搞定 Agent 训练数据与流程

Data recipes and robust infrastructure for training AI agents

★ 310 2026-10-17
ParScale 开源

并行跑多个模型,不堆参数也能提升性能

Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scali···

★ 486 2026-08-09
CAME 开源

减少大模型训练显存,优化器也能自适应省内存

[ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory O···

★ 100 2026-08-09
OneTrainer 开源

图形化界面搞定扩散模型训练,从 LoRA 到全量微调一站式完成。

OneTrainer is a one-stop solution for all your Diffusion training needs.

★ 3264 2026-08-10
LLM-ESR 开源

让大模型推理步步为营,减少中间错误,提升最终答案准确率。

[NeurIPS'24 Spotlight] The official implementation code of LLM-ESR.

★ 54 2026-08-09