Meta AI SAM demo配置安装

目录

1.简单介绍

2.系统环境

3.安装配置

4.运行demo

4.1 fork

4.2修改配置


1.简单介绍

Meta发布首个基础图像分割模型SAM(SegmentAnythingModel)。

SAM已经学会了关于物体的一般概念,并且它可以为任何图像或视频中的任何物体生成掩码,甚至包括在训练过程中没有遇到过的物体和图像类型。

SAM足够通用,可以涵盖广泛的用例,并且可以在新的图像领域上即开即用,无需额外的训练。

SAM 可以使用各种输入提示包括点击,框选和文字,指定要在图像中分割的内容,并且可以灵活集成其他系统。

SAM 初步验证了多模态技术路径及其泛化能力,相当于计算机视觉领域的GPT-3。

官网:

Segment Anything | Meta AI

github:

facebookresearch/segment-anything

官网可以在线体验,以下内容主要介绍配置py调用sam的过程;

2.系统环境

没有GPU跑不了;没有GPU跑不了;没有GPU跑不了;

笔者配置:windows10,GPU 6G;

cmd查看显存占用情况;

nvidia-smi

### 关于 SAMDemo 和 示例代码 对于 Segment Anything Model (SAM),其官方团队已经提供了多个可用的演示版本以及开源代码,方便开发者快速上手并测试该模型的功能[^2]。以下是有关 SAM 的具体资源和使用方法: #### 官方 GitHub 仓库 Meta AI 提供了一个完整的 Python 实现库 `segment-anything`,其中包含了 SAM 的预训练权重文件以及详细的文档说明。可以通过以下命令安装依赖项: ```bash pip install git+https://github.com/facebookresearch/segment-anything.git ``` 此外,在该项目的 README 文件中还提供了一些简单的示例脚本用于加载模型并对图片执行分割操作。 #### Colab 在线体验版 为了降低用户的入门门槛,社区贡献者基于原始项目开发出了多种交互式的 Notebook 形式工具链(例如 Google Colab 笔记本)。这些在线平台允许任何人无需本地配置环境即可运行 SAM 幃算法实例化过程及其基本功能验证环节。 下面是一个典型的调用流程展示片段: ```python from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor import torch device = 'cuda' if torch.cuda.is_available() else 'cpu' # 加载模型 sam_checkpoint = "path/to/sam_vit_h_4b8939.pth" model_type = "vit_h" sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device) predictor = SamPredictor(sam) image = cv2.imread('example.jpg') predictor.set_image(image) input_point = np.array([[500, 375]]) input_label = np.array([1]) masks, scores, logits = predictor.predict( point_coords=input_point, point_labels=input_label, multimask_output=True, ) ``` 上述代码展示了如何通过指定兴趣区域内的单个像素点来获取对应目标对象的掩码信息[^3]。 #### 社区扩展应用案例分享 除了基础 API 外,还有不少第三方创作者围绕 SAM 构建起了更加友好的图形界面程序,比如结合 LabelImg 使用从而达成更高效的半自动化标注效果[^1];亦或是利用 Streamlit 创建 Web 应用服务端口使得远程访问成为可能等等。 ---
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值