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











评论