Segment Anything
一、项目简介
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)
五、注意事项
- 需下载权重文件,体积较大;
- 大模型推理需一定显存;
- 注意许可与生成内容的合规使用。