这个项目能做什么
SamGeo(segment-geospatial)是一个用于地理空间数据分割的Python包,基于Segment Anything Model (SAM)。它受Aliaksandr Hancharenka的segment-anything-eo仓库启发并改编,旨在以最少编码使SAM适用于地理空间分析。该包以MIT许可证发布,文档见samgeo.gishub.org,并发表于《Journal of Open Source Software》。
README中描述的功能包括:
- 从Tile Map Service (TMS)服务器下载地图瓦片并构建GeoTIFF文件。
- 使用SAM和HQ-SAM分割GeoTIFF文件。
- 使用文本提示分割遥感影像。
- 交互式创建前景和背景标记,或从矢量数据集加载现有标记。
- 将分割结果保存为GeoPackage、Shapefile或GeoJSON,并将输入提示保存为GeoJSON。
- 在交互式地图上可视化分割结果。
- 从时间序列遥感影像中分割对象。
- 通过REST API通过HTTP提供分割服务。
模型变体和安装:该包支持SAMGeo 1、2和3,Fast SAM,HQ-SAM,用于文本提示的Grounding DINO,以及特征边缘重建算法。安装选项包括pixi(推荐用于复杂的PyTorch/CUDA依赖)、PyPI(带有samgeo、samgeo2、samgeo3、fast、hq、text、fer、api、all和extra等附加选项)以及conda-forge。针对SAM 3有单独的Windows程序文档,包括triton-windows。
集成和生态系统:提供了QGIS插件,README指向ArcGIS工具箱和ArcGIS Pro示例。示例涵盖卫星影像、自动掩码生成、输入/框/文本提示、批量文本提示分割、游泳池、Maxar Open Data影像和ArcGIS Pro。视频教程链接自作者的YouTube频道。
硬件和法律说明:SAM计算密集;建议使用至少8 GB内存的GPU,并建议使用免费的Google Colab GPU资源或AWS研究积分。README声明该仓库仅用于教育目的,用户自行承担下载瓦片的风险,大规模瓦片下载可能需要底图提供商的许可。该项目感谢NASA资助和AWS支持,并致谢上游开源项目,包括segment-anything、SAM 3、segment-anything-eo、tms2geotiff、GroundingDINO和lang-segment-anything。
评论
0 人表达喜爱 · 满10人后显示 Deer Point
登录后参与讨论。