TorchgeoPytorch 地理空间数据库

Torchgeo 是微软开源的一个 PyTorch 域的库，提供特定于地理空间数据的数据集、转换、采样器和预训练模型，有点类似于 torchvision。

此库主要应用于两方面：

• 方便机器学习专家在工作流程中使用地理空间数据
• 方便遥感专家在机器学习工作流程中使用他们的数据。

可以参阅安装说明、文档和示例，以了解如何使用 torchgeo。

示范用例

• 使用基于 PyTorch Lightning 的训练脚本对模型进行训练和测试

$ python train.py config_file=conf/landcoverai.yaml

• 下载并使用热带气旋风力估算竞赛的数据集

该数据集来自 Driven Data 与 Radiant Earth 合作举办的比赛，在 torchgeo 中使用这个数据集非常简单：

import torchgeo.datasets

dataset = torchgeo.datasets.TropicalCycloneWindEstimation(split="train", download=True)
print(dataset[0]["image"].shape)
print(dataset[0]["label"])