位置: IT常识 - 正文

【图像分割】Meta分割一切(SAM)模型环境配置和使用教程(图像分割最新算法)

编辑:rootadmin
【图像分割】Meta分割一切(SAM)模型环境配置和使用教程

推荐整理分享【图像分割】Meta分割一切(SAM)模型环境配置和使用教程(图像分割最新算法),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:图像分割ncut,图像分割csdn,图像分割unet,图像分割实现,图像分割实现,图像分割otsu,图像分割otsu,图像分割miou,内容如对您有帮助,希望把文章链接给更多的朋友!

注意:python>=3.8, pytorch>=1.7,torchvision>=0.8

Feel free to ask any question. 遇到问题欢迎评论区讨论.

官方教程:

https://github.com/facebookresearch/segment-anything1 环境配置1.1 安装主要库:

(1)pip:

有可能出现错误,需要配置好Git。

pip install git+https://github.com/facebookresearch/segment-anything.git

(2)本地安装:

有可能出现错误,需要配置好Git。

git clone git@github.com:facebookresearch/segment-anything.gitcd segment-anything; pip install -e .

(3)手动下载+手动本地安装:

 zip文件:

链接:https://pan.baidu.com/s/1dQ--kTTJab5eloKm6nMYrg提取码:1234

解压后运行: 

cd segment-anything-mainpip install -e .1.2 安装依赖库:pip install opencv-python pycocotools matplotlib onnxruntime onnx

matplotlib 3.7.1和3.7.0可能报错

如果报错:pip install matplotlib==3.6.2

1.3 下载权重文件:

下载三个权重文件中的一个,我用的第一个。

default or vit_h: ViT-H SAM model.vit_l: ViT-L SAM model.vit_b: ViT-B SAM model.

 如果下载过慢:

链接:https://pan.baidu.com/s/11wZUcjYWNL6kxOH5MFGB-g 提取码:1234 2 使用教程2.1 根据在图片上选择的点扣出物体【图像分割】Meta分割一切(SAM)模型环境配置和使用教程(图像分割最新算法)

原始图像:

 导入依赖库和展示相关的函数:

import cv2import matplotlib.pyplot as pltimport numpy as npfrom segment_anything import sam_model_registry, SamPredictordef show_mask(mask, ax, random_color=False): if random_color: color = np.concatenate([np.random.random(3), np.array([0.6])], axis=0) else: color = np.array([30 / 255, 144 / 255, 255 / 255, 0.6]) h, w = mask.shape[-2:] mask_image = mask.reshape(h, w, 1) * color.reshape(1, 1, -1) ax.imshow(mask_image)def show_points(coords, labels, ax, marker_size=375): pos_points = coords[labels == 1] neg_points = coords[labels == 0] ax.scatter(pos_points[:, 0], pos_points[:, 1], color='green', marker='*', s=marker_size, edgecolor='white', linewidth=1.25) ax.scatter(neg_points[:, 0], neg_points[:, 1], color='red', marker='*', s=marker_size, edgecolor='white', linewidth=1.25)

确定使用的权重文件位置和是否使用cuda等:

sam_checkpoint = "F:\sam_vit_h_4b8939.pth"device = "cuda"model_type = "default"

模型实例化:

sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)sam.to(device=device)predictor = SamPredictor(sam)

读取图像并选择抠图点:

image = cv2.imread(r"F:\Dataset\Tomato_Appearance\Tomato_Xishi\images\xs_1.jpg")image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)predictor.set_image(image)input_point = np.array([[1600, 1000]])input_label = np.array([1])plt.figure(figsize=(10,10))plt.imshow(image)show_points(input_point, input_label, plt.gca())plt.axis('on')plt.show()

 扣取图像(会同时提供多个扣取结果):

masks, scores, logits = predictor.predict( point_coords=input_point, point_labels=input_label, multimask_output=True,)# 遍历读取每个扣出的结果for i, (mask, score) in enumerate(zip(masks, scores)): plt.figure(figsize=(10,10)) plt.imshow(image) show_mask(mask, plt.gca()) show_points(input_point, input_label, plt.gca()) plt.title(f"Mask {i+1}, Score: {score:.3f}", fontsize=18) plt.axis('off') plt.show()

     

 尝试扣取其他位置:

 

2.2 扣取图像中的所有物体

官方教程:

https://github.com/facebookresearch/segment-anything/blob/main/notebooks/automatic_mask_generator_example.ipynb

依赖库和函数导入:

from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictorimport cv2import matplotlib.pyplot as pltimport numpy as npdef show_anns(anns): if len(anns) == 0: return sorted_anns = sorted(anns, key=(lambda x: x['area']), reverse=True) ax = plt.gca() ax.set_autoscale_on(False) polygons = [] color = [] for ann in sorted_anns: m = ann['segmentation'] img = np.ones((m.shape[0], m.shape[1], 3)) color_mask = np.random.random((1, 3)).tolist()[0] for i in range(3): img[:,:,i] = color_mask[i] ax.imshow(np.dstack((img, m*0.35)))

读取图片:

image = cv2.imread(r"F:\Dataset\Tomato_Appearance\Tomato_Xishi\images\xs_1.jpg")image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

实例化模型:

sam_checkpoint = "F:\sam_vit_h_4b8939.pth"model_type = "default"device = "cuda"sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)sam.to(device=device)

 分割并展示(速度有点慢):

mask_generator = SamAutomaticMaskGenerator(sam)masks = mask_generator.generate(image)plt.figure(figsize=(20,20))plt.imshow(image)show_anns(masks)plt.axis('off')plt.show()

2.3 根据文字扣取物体

配置另外一个库:

https://github.com/IDEA-Research/Grounded-Segment-Anything

后续更新细节

本文链接地址:https://www.jiuchutong.com/zhishi/288855.html 转载请保留说明!

上一篇:【Java基础】一个Java文件可以有多个类(外部类、内部类)(java基础教程)

下一篇:被翡翠湾环绕的芬尼特岛,太浩湖,加利福尼亚 (© Rachid Dahnoun/Tandem Stills + Motion)(翡翠湾攻略)

  • vivox80怎么去除照片水印(vivox80怎么去除拍照水印)

    vivox80怎么去除照片水印(vivox80怎么去除拍照水印)

  • qq付费音乐包和绿钻区别(qq付费音乐包和绿钻的区别)

    qq付费音乐包和绿钻区别(qq付费音乐包和绿钻的区别)

  • 华为nova4后屏是什么材质(华为nova4后屏碎了,容易换吗)

    华为nova4后屏是什么材质(华为nova4后屏碎了,容易换吗)

  • 微信头像更换影响收款吗(微信头像更换影响运气吗)

    微信头像更换影响收款吗(微信头像更换影响运气吗)

  • 苹果手机xs max通话怎么录音

    苹果手机xs max通话怎么录音

  • 物联网卡不销户有什么影响(物联卡不销卡有影响吗)

    物联网卡不销户有什么影响(物联卡不销卡有影响吗)

  • p40 pro pro+区别(p40 pro和pro+的区别)

    p40 pro pro+区别(p40 pro和pro+的区别)

  • 苹果7p微信老是闪退(苹果7p微信老是闪退咋回事)

    苹果7p微信老是闪退(苹果7p微信老是闪退咋回事)

  • 此系统不符合安装该软件最低要求(此系统不符合安装该软件需求)

    此系统不符合安装该软件最低要求(此系统不符合安装该软件需求)

  • 手机卡三个月不交费会自动注销吗(手机卡三个月不用就自动注销了吗)

    手机卡三个月不交费会自动注销吗(手机卡三个月不用就自动注销了吗)

  • 得物极速发货多少天到手(得物极速发货多少钱)

    得物极速发货多少天到手(得物极速发货多少钱)

  • opporeno为什么录不上指纹了(opporeno为什么录屏没有声音)

    opporeno为什么录不上指纹了(opporeno为什么录屏没有声音)

  • 苹果白点怎么取消(苹果白点里面功能怎么调)

    苹果白点怎么取消(苹果白点里面功能怎么调)

  • 微信注册需要辅助验证?(微信注册需要辅助验证的条件是什么)

    微信注册需要辅助验证?(微信注册需要辅助验证的条件是什么)

  • b站创作激励以前的视频算吗(b站创作激励以前的视频怎么处理)

    b站创作激励以前的视频算吗(b站创作激励以前的视频怎么处理)

  • xp是32位还是64位

    xp是32位还是64位

  • 抖音直播很卡是什么原因(抖音直播卡是网络原因还是手机)

    抖音直播很卡是什么原因(抖音直播卡是网络原因还是手机)

  • 网易云音乐hd是什么意思(网易云音乐hd是苹果专用的吗)

    网易云音乐hd是什么意思(网易云音乐hd是苹果专用的吗)

  • vivoz1i支持多少w快充(vivoz1支持多少瓦)

    vivoz1i支持多少w快充(vivoz1支持多少瓦)

  • oppo手机呼吸灯在哪(oppo手机呼吸灯颜色在哪里调节)

    oppo手机呼吸灯在哪(oppo手机呼吸灯颜色在哪里调节)

  • 台式键盘fn键在哪(台式键盘fn键在哪个位置)

    台式键盘fn键在哪(台式键盘fn键在哪个位置)

  • opsson是什么牌子手机(opsson是什么牌子手机多少钱)

    opsson是什么牌子手机(opsson是什么牌子手机多少钱)

  • 小米商城碎屏险怎么买(小米商城碎屏险和支付宝碎屏险)

    小米商城碎屏险怎么买(小米商城碎屏险和支付宝碎屏险)

  • 小米8如何找回隐藏视频(小米8如何找回以现在的应用)

    小米8如何找回隐藏视频(小米8如何找回以现在的应用)

  • 完美解决 Error: Cannot find module ‘@vue/cli-plugin-eslint‘ 报错(完美解决战网已休眠正在唤醒它)

    完美解决 Error: Cannot find module ‘@vue/cli-plugin-eslint‘ 报错(完美解决战网已休眠正在唤醒它)

  • 一次性医用外科口罩哪个牌子好
  • 设备检验检测
  • 利息税怎么算的
  • 投资收益科目在贷方
  • 投资性房地产递延所得税资产怎么算
  • 办理企业所得税退税
  • 预缴增值税 已交税金
  • 土地前期开发费用计征契税
  • 收到跨境分红的会计处理怎么做?
  • 被开除了怎么办理手续
  • 非公益性捐赠支出是永久性差异吗
  • 专用设备抵免企业所得税目录
  • 购销合同印花税优惠政策
  • 混合销售举例说明
  • 汽车销售公司买车后续维修保养
  • 增值税电子发票没有发票专用章
  • 其他收益与营业外收入的区别
  • 月末计提短期借贷利息
  • 收到汇票怎么承兑
  • 路由器管家怎么登录
  • 新会计准则下企业合并财务报表问题研究
  • 企业奠基费用如何入账
  • vim diff对比
  • PHP:session_set_save_handler()的用法_Session函数
  • 购买材料时采购会计分录
  • 当月认证抵扣的进项税发票一定要入帐做成本吗?
  • vue项目中使用opencv
  • 猿人知乎
  • 可以享受企业所得税加计扣除的有
  • php的判断语句
  • 定额发票过期时间是多久
  • Windows下php+mysql5.7配置教程
  • 固定资产一次性折旧账务处理
  • 冲账怎么写?
  • 原材料当废品卖出分录
  • 前端get请求传数组
  • java单点登录token
  • 反射dll
  • 应交税费如何调整
  • 应收账款质押融资平台
  • 交上年企业所得税多少
  • 民营医院发票样式
  • 各会计科目的含义
  • python中字符串与数字相乘的结果
  • python创建ndarray
  • 非盈利机构怎么说
  • 预收款销售商品如何与业务员对账
  • 从合同有什么
  • SQL Server 2008 R2 应用及多服务器管理
  • 法定盈余公积的主要用途
  • 银行汇票与商业汇票的本质不同
  • 在建工程领用原材料税费怎么处理
  • 对于相关人员培训存在的问题
  • 小规模纳税人购进税控收款机
  • 住房基金可以取出来用吗
  • 原材料用于在建工程
  • 扣缴个人所得税报告表
  • 投资其他公司做什么科目
  • 会计档案销毁方案怎么写
  • 财务会计是学什么课程
  • 做好一个服务器工作
  • 重装系统要注意哪些
  • 如何关闭windows防护
  • linux 压缩包zip
  • win批处理命令
  • win7系统硬盘
  • windowsxp资源管理器在哪里
  • centos6启动服务的命令
  • linux文件目录的含义
  • win8正版系统自带
  • win8跳过开机密码
  • 如何在win7电脑上添加新的打印机
  • win8如何激活
  • javascript的字符串操作
  • JavaScript事件类型中焦点、鼠标和滚轮事件详解
  • javascript中继承
  • HTTP之Range理解
  • 重庆办理4050社保需要什么材料
  • 税务上门核查要看什么
  • 公司财务专用章的用途
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设