← 返回 AI 开源导航

Segment Anything

图像生成协议 Apache-2.0★ 55.1kJupyter Notebook

一、项目简介

Segment Anything(SAM)是 Meta 开源的通用图像分割模型。它能通过点、框或文本等提示,对任意图像生成高质量分割掩码,是计算机视觉领域的里程碑,广泛用于数据标注、图像编辑与视觉管线。

二、核心特性

  • 零样本分割:无需针对新类别训练;
  • 多提示方式:点、框、掩码、文本;
  • 高质量掩码:边界精细;
  • 可提示分割任务(promptable segmentation);
  • 轻量变体:SAM 2、MobileSAM 等;
  • 生态集成广泛。

三、适用场景

  • 图像数据标注与预处理;
  • 抠图、图像编辑与合成;
  • 视觉管线中的通用分割步骤。

四、快速上手

pip install segment-anything
from segment_anything import sam_model_registry, SamPredictor
sam = sam_model_registry["vit_b"](checkpoint="sam_vit_b.pth")
predictor = SamPredictor(sam)

五、注意事项

  • 需下载权重文件,体积较大;
  • 大模型推理需一定显存;
  • 注意许可与生成内容的合规使用。