Segment Anything图像分割模型

Segment Anything Model（SAM）从输入提示中产生高质量的物体遮罩，它可以用来为图像中的所有物体产生遮罩。它已经在一个由 1100 万张图像和 11 亿个遮罩组成的数据集上进行了训练，并在各种分割任务中具有强大的性能。

安装

需要 Python >= 3.8，以及 PyTorch >= 1.7 和 TorchVision >= 0.8。请按照这里的说明来安装PyTorch和TorchVision这两个依赖项。我们强烈建议同时安装支持CUDA的PyTorch和TorchVision。

安装 Segment Anything:

pip install git+https://github.com/facebookresearch/segment-anything.git

或者在本地克隆仓库并安装

git clone git@github.com:facebookresearch/segment-anything.git
cd segment-anything; pip install -e .

以下是遮罩后处理、以 COCO 格式保存遮罩、实例 notebooks 和以 ONNX 格式导出模型所需的可选依赖。运行实例 notebooks 也需要 jupyter。

pip install opencv-python pycocotools matplotlib onnxruntime onnx