TorchGeo — PyTorch训练空间数据开源领域库

TorchGeo 是一个 PyTorch 领域库，类似于 torchvision，专门提供地理空间数据的数据集、采样器、变换和预训练模型。可以帮助机器学习专家轻松处理地理空间数据，也能够让遥感领域的专家探索机器学习解决方案。

核心功能如下：

地理空间数据集和采样器

1. 超过 20 种卫星图像和掩码的数据加载器（列表见这里：https://torchgeo.readthedocs.io/en/stable/api/datasets.html#geospatial-datasets）
2. 自动坐标参考系统（CRS）重投影和重采样
3. 支持多模态学习和数据融合

基准数据集

• 超过 50 种基准数据集加载器（见这里：https://torchgeo.readthedocs.io/en/stable/api/datasets.html#non-geospatial-datasets）
• 快速试验各种模型
• 自动下载大多数数据集

预训练权重

• 超过 40 种在地理空间数据上预训练的模型（详细模型见这里：https://torchgeo.readthedocs.io/en/stable/api/models.html#pretrained-weights）
• 原生支持多光谱图像
• 使小数据集上的迁移学习成为可能

使用 Lightning 实现可重复性

• 内置训练/验证/测试拆分和数据增强
• 分类、回归、检测、分割
• 命令行界面，支持 YAML 文件

官方地址

项目地址：

为了防止资源恶意爬取导致被和谐，此处内容被作者隐藏。

验证码：

关注本站微信公众号，回复“资源下载”，获取验证码。

在微信里搜索“麻辣GIS”或微信扫描右侧二维码即可关注本站微信公众号。

PS：目前本项目文档本站暂未提供汉化版本。