terratorch

收录理由

TerraTorch 是一套基于 PyTorch Lightning 的地理空间基础模型微调工具包，它的设计目标是让研究人员不必在数据管线与训练流程上反复折腾。你可以将 Prithvi、TerraMind、Granite 以及 TorchGeo、timm 中的各种骨干网络与可插拔的解码器自由组合，用 Python 或 YAML 配置就能定义语义分割、回归或分类任务，而不用手写训练循环。由于底层依托 TorchGeo，波段处理、多时相输入和数据加载这些头疼的问题已被妥善解决，这套框架还能延伸到太阳物理和天气预报场景。实际使用时，应用工程师只需一份配置文件，就能从预训练检查点出发，微调出用于火烧迹地或作物类型提取的模型。它更像一个供你在此基础上构建的框架，而非现成的模型仓库，适合希望掌控训练与推理细节、同时把重复劳动压缩到极致的团队。