Semantic Segmentation with SAM and GeoAI¶

Install package¶

To use the geoai-py package, ensure it is installed in your environment. Uncomment the command below if needed.

In [ ]:

# %pip install geoai-py

# %pip install geoai-py

Import libraries¶

In [ ]:

import geoai
from geoai.sam import SamGeo

import geoai from geoai.sam import SamGeo

Download sample data¶

In [ ]:

url = (
    "https://huggingface.co/datasets/giswqs/geospatial/resolve/main/sam_demo_image.tif"
)
image = geoai.download_file(url)

url = ( "https://huggingface.co/datasets/giswqs/geospatial/resolve/main/sam_demo_image.tif" ) image = geoai.download_file(url)

Visualize data¶

In [ ]:

geoai.view_raster(url)

geoai.view_raster(url)

Initialize model¶

In [ ]:

sam = SamGeo(model="facebook/sam-vit-huge", automatic=True)

sam = SamGeo(model="facebook/sam-vit-huge", automatic=True)

Automatic mask generation¶

In [ ]:

sam.generate(source=image, output="masks.tif", foreground=True)

sam.generate(source=image, output="masks.tif", foreground=True)

Visualize results¶

In [ ]:

sam.show_masks()

sam.show_masks()

In [ ]:

sam.show_anns(alpha=0.6)

sam.show_anns(alpha=0.6)

In [ ]:

geoai.view_raster("masks.tif", colormap="tab20", opacity=0.6, basemap=url)

geoai.view_raster("masks.tif", colormap="tab20", opacity=0.6, basemap=url)

In [ ]:

geoai.create_split_map(
    left_layer="masks.tif",
    right_layer=url,
    left_args={
        "colormap": "tab20",
        "opacity": 1,
    },
)

geoai.create_split_map( left_layer="masks.tif", right_layer=url, left_args={ "colormap": "tab20", "opacity": 1, }, )