torchgeo

收录理由

做过卫星或航拍影像视觉模型的人，多半都体会过把地理空间数据塞进 PyTorch 流程里的那种折腾。TorchGeo 就是冲着消除这层麻烦来的：作为 PyTorch 的领域库，它效仿 torchvision 的思路，为上百个地理空间数据集提供带坐标系感知的加载器，采样器能挑出有地理意义的窗口，还自带多光谱变换以及 SeCo、SatMAE、DOFA、Prithvi 等骨干网络的预训练权重。用这套工具，可以直接读入 13 波段的 Sentinel-2 数据，不必再临时拼凑 RGB 方案，各个组件也能无缝接进标准的 PyTorch 和 Lightning 训练循环。对于土地覆盖分类、建筑物轮廓语义分割或目标检测这类任务，它把数据管线和现成检查点都备好了，上手就能用。无论是刚接触地理空间格式的机器学习工程师，还是想用标准深度学习流程、不想重造轮子的遥感专家，都能从中找到顺手的地方。