TorchGeo — PyTorch训练空间数据开源领域库
发布时间: 2024-09-02
所属分类: 开源GIS
TorchGeo 是一个 PyTorch 领域库,类似于 torchvision,专门提供地理空间数据的数据集、采样器、变换和预训练模型。可以帮助机器学习专家轻松处理地理空间数据,也能够让遥感领域的专家探索机器学习解决方案。
核心功能如下:
地理空间数据集和采样器
- 超过 20 种卫星图像和掩码的数据加载器(列表见这里:https://torchgeo.readthedocs.io/en/stable/api/datasets.html#geospatial-datasets)
- 自动坐标参考系统(CRS)重投影和重采样
- 支持多模态学习和数据融合
基准数据集
- 超过 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:目前本项目文档本站暂未提供汉化版本。
相关阅读
声明
1.本文所分享的所有需要用户下载使用的内容(包括但不限于软件、数据、图片)来自于网络或者麻辣GIS粉丝自行分享,版权归该下载资源的合法拥有者所有,如有侵权请第一时间联系本站删除。
2.下载内容仅限个人学习使用,请切勿用作商用等其他用途,否则后果自负。
手机阅读
公众号关注
知识星球
手机阅读
最新GIS干货
私享圈子