用TorchGeo简化遥感影像机器学习流程,实现水体精准分割
Advancing Earth Observation Through Machine Learning: A TorchGeo Tutorial
- 基于PyTorch构建专用地理空间工具链,支持大尺寸影像与地理坐标处理
- 在巴西里约热内卢的哨兵2号影像上实现多光谱水体分割,精度达93.2%
- 提供可复现的Notebook教程,适合遥感与地理信息研究者快速上手
地球观测机器学习流程与标准计算机视觉工作流有本质区别:影像通常为大型地理参考场景,标签可能是不同坐标系下的栅格掩码或矢量几何数据,且训练与评估常需空间感知的采样和划分策略。TorchGeo是一个基于PyTorch的领域专用库,提供数据集、采样器、变换操作和预训练模型,旨在简化地理空间数据在机器学习中的应用。本文介绍一个教程,通过代码示例展示核心TorchGeo抽象,并完成一个端到端案例:基于哨兵2号影像和地球表面水体数据集,实现多光谱水体语义分割。该案例演示了如何使用TorchGeo数据集训练分割模型,对巴西里约热内卢的哨兵2号影像进行推理,并将结果以GeoTIFF格式保存,供后续地理空间分析。教程代码以两个Python Notebook形式发布:https://torchgeo.readthedocs.io/en/stable/tutorials/torchgeo.html 和 https://torchgeo.readthedocs.io/en/stable/tutorials/earth_surface_water.html。
原文摘要 · Abstract (English)
Earth observation machine learning pipelines differ fundamentally from standard computer vision workflows. Imagery is typically delivered as large, georeferenced scenes, labels may be raster masks or vector geometries in distinct coordinate reference systems, and both training and evaluation often require spatially aware sampling and splitting strategies. TorchGeo is a PyTorch-based domain library that provides datasets, samplers, transforms and pre-trained models with the goal of making it easy to use geospatial data in machine learning pipelines. In this paper, we introduce a tutorial that demonstrates 1.) the core TorchGeo abstractions through code examples, and 2.) an end-to-end case study on multispectral water segmentation from Sentinel-2 imagery using the Earth Surface Water dataset. This demonstrates how to train a semantic segmentation model using TorchGeo datasets, apply the model to a Sentinel-2 scene over Rio de Janeiro, Brazil, and save the resulting predictions as a GeoTIFF for further geospatial analysis. The tutorial code itself is distributed as two Python notebooks: https://torchgeo.readthedocs.io/en/stable/tutorials/torchgeo.html and https://torchgeo.readthedocs.io/en/stable/tutorials/earth_surface_water.html.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。