comfyui_segment_anything

admin2024-09-14423

基于GroundingDino和SAM,使用语义字符串对图像中的任何元素进行分割。sd-webui-segment-anything的comfyui版本。|常用场景,就是语义控制生成蒙版,用途广泛,实用!

插件下载

image.png

地址:https://github.com/storyicon/comfyui_segment_anything

模型安装

1、bert-base-uncased

        下载地址:https://huggingface.co/bert-base-uncased/tree/main

image.png

        以上5个文件下载后放在models/bert-base-uncased目录。

2、GroundingDino

image.png

    文件下载后放置在models/grounding-dino目录。

3、SAM

image.png

文件下载后放在models/sams目录。

SAM模型的不同版本主要区别在于模型的大小、性能和适用场景‌。以下是具体说明:

  • sam_vit_b‌:基础版本,拥有最小的神经网络规模,但在性能上并不落后。

  • sam_vit_l‌:大型版本,拥有更多的参数,性能有所提升,提供最佳的平衡。

  • sam_vit_h‌:巨型版本,参数最多,准确性最高,但速度较慢。

  • sam_hq_vit_h模型名字中的hq代表“高质量”(High Quality)‌。

  • mobile_sam‌:针对移动设备优化的轻量级版本,通过解耦蒸馏法减少参数量,提高推理速度,适合在资源受限的设备上部署。

基础用法

image.png

节点

SAM模型加载器

G-Dino模型加载器

G-DinoSAM语义分割

反转遮罩

遮罩是否存在


网友评论