SegmentAnythingModel(SAM)从输入提示中产生高质量的物体遮罩,它可以用来为图像中的所有物体产生遮罩。它已经在一个由1100万张图像和11亿个遮罩组成的数据集上进行了训练,并在各种分割任务中具有强大的性能。
安装需要Python>=3.8,以及PyTorch>=1.7和 TorchVision>=0.8。请按照这里的说明来安装PyTorch和TorchVision这两个依赖项。我们强烈建议同时安装支持CUDA的PyTorch和TorchVision。
安装SegmentAnything:
pipinstallgit+https://github.com/facebookresearch/segment-anything.git或者在本地克隆仓库并安装
gitclonegit@github.com:facebookresearch/segment-anything.gitcdsegment-anything;pipinstall-e.以下是遮罩后处理、以COCO格式保存遮罩、实例notebooks和以ONNX格式导出模型所需的可选依赖。运行实例notebooks也需要jupyter。
pipinstallopencv-pythonpycocotoolsmatplotlibonnxruntimeonnx
评论