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
评论