rsrscoder / segment-geospatial

A Python package for segmenting geospatial data with the Segment Anything Model (SAM)

segment-geospatial

A Python package for segmenting geospatial data with the Segment Anything Model (SAM)

The segment-geospatial package draws its inspiration from segment-anything-eo repository authored by Aliaksandr Hancharenka. To facilitate the use of the Segment Anything Model (SAM) for geospatial data, I have developed the segment-anything-py and segment-geospatial Python packages, which are now available on PyPI and conda-forge. My primary objective is to simplify the process of leveraging SAM for geospatial data analysis by enabling users to achieve this with minimal coding effort. I have adapted the source code of segment-geospatial from the segment-anything-eo repository, and credit for its original version goes to Aliaksandr Hancharenka.

• Free software: MIT license
• Documentation: https://samgeo.gishub.org

Features

• Download map tiles from Tile Map Service (TMS) servers and create GeoTIFF files
• Segment GeoTIFF files using the Segment Anything Model (SAM)
• Save segmentation results as common vector formats (GeoPackage, Shapefile, GeoJSON, etc.)
• Visualize segmentation results on interactive maps

Examples

• Segmenting satellite imagery
• Automatically generating object masks

Demos

• Automatic mask generator

Acknowledgements

This package was made possible by the following open source projects. Credit goes to the developers of these projects.

• segment-anything
• segment-anything-eo
• tms2geotiff