- 安装库
# %pip install segment-geospatial
- 包的引入
import os
import leafmap
from samgeo import SamGeo, show_image, download_file, overlay_images, tms_to_geotiff
- 替换成自己的遥感图像
image = '/content/181.tif'
- 模型调用
sam = SamGeo(
model_type="vit_h",
sam_kwargs=None,
)
- 生成分割后的图像
sam.generate(image, output="masks.tif", foreground=True, unique=True)