Segment Anything 图像分割模型开源项目

我要开发同款
匿名用户2023年04月06日
39阅读
开发技术Python
授权协议Apache-2.0

作品详情

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
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论