Stable Diffusion - 扩展 SegmentAnything 和 GroundingDINO 实例分割算法插件的配置与使用

欢迎关注我的CSDN：https://spike.blog.csdn.net/

本文地址：https://blog.csdn.net/caroline_wendy/article/details/131918652

Paper and GitHub：

Segment Anything: SAM - Segment Anything
- GitHub: https://github.com/facebookresearch/segment-anything
Grounding DINO: Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection
- 定位 DINO: 结合DINO与基于语言的预训练用于开放集合的目标检测
- GitHub: https://github.com/IDEA-Research/GroundingDINO

Segment Anything:

Segment Anything 是关于图像分割领域的研究，提出了一个新的任务、模型和数据集，能够根据输入的提示（如点或框）生成高质量的物体掩码，或者对于整张图像进行分割。这个模型被称为 Segment Anything Model (SAM)，使用了一个高效的模型结构，结合卷积神经网络和 Transformer 架构。SAM在一个由 11M 张图像和 1.1B 个掩码组成的大规模数据集上进行了训练，这个数据集被称为 SA-1B。SAM 具有强大的零样本迁移能力，能够适应不同的图像分布和任务。在多个分割任务上评估 SAM 的性能，发现零样本表现令人印象深刻，甚至超过了之前的全监督结果。

SAM的主要贡献：

提出了一个新的图像分割任务，即 Segment Anything (SA)，要求模型根据输入的提示（如点或框）生成物体掩膜，或者对整张图像进行分割。
设计并实现一个高效且强大的图像分割模型，即 Segment Anything Model (SAM)，使用基于 ResNet-50 的卷积神经网络作为编码器，和基于 ViT-B/16 的 Transformer 作为解码器。
利用 SAM 在一个数据收集循环中，构建目前最大的图像分割数据集（远远超过之前的数据集），即 SA-1B，包含了 11M 张经过许可和隐私保护的图像，以及 1.1B 个物体掩码。
评估 SAM 在多个分割任务上的零样本迁移能力，包括 COCO、ADE20K、Cityscapes、Mapillary Vistas、Open Images V6等，发现在所有任务上都取得了优异的结果，甚至超过了之前的全监督方法。

Grounding DINO:

Grounding DINO 是开放集合的目标检测器，通过结合基于 Transformer 的检测器 DINO 和基于语言的预训练模型，可以检测出任意人工输入的类别名字或者相关表达的目标。

开放集合的目标检测是指在没有给定类别标签的情况下，根据人类的语言输入，识别出图像中的目标，并且给出其位置和类别信息。这是一个具有挑战性和实用性的任务，可以应用于多种场景，比如搜索引擎、智能助理、无人驾驶等。然而，现有的目标检测器通常是基于有限的类别集合进行训练和测试的，也就是说，只能检测出预先定义好的类别，而不能处理新颖或未知的类别。为了解决这个问题，Grounding DINO 有效融合语言和视觉模态，使得检测器可以根据语言输入来扩展其概念范围，并且可以处理多种形式的语言输入，比如类别名字、属性描述、指代表达等。

具体来说，Grounding DINO 采用 DINO 作为基础的目标检测器，DINO 是一种基于Transformer的端到端的目标检测器，不需要锚框或者预定义的特征金字塔，而是使用可学习的查询向量来表示目标，并且使用自注意力机制来捕捉图像中的全局上下文信息。为了使 DINO 能够处理开放集目标检测任务，Grounding DINO 提出 3 个关键的改进点：

特征增强器（Feature Enhancer）：这是一个基于 Transformer 的编码器，将图像特征和语言特征进行融合，并且输出一个增强后的图像特征表示。这样可以使得图像特征包含更多与语言输入相关的信息，从而提高检测器对新颖或未知类别的泛化能力。
语言引导查询选择（Language-Guided Query Selection）：这是一个基于 Transformer 的解码器，将增强后的图像特征和语言特征作为输入，并且输出一个查询向量集合。这个查询向量集合可以根据语言输入来动态地调整其数量和内容，从而更好地匹配图像中与语言输入相关的目标。
跨模态解码器（Cross-Modality Decoder）：这是一个基于 Transformer 的解码器，将查询向量集合和增强后的图像特征作为输入，并输出最终的检测结果。这个解码器可以利用自注意力机制来进行跨模态融合，并且可以使用多头注意力机制来进行多尺度特征融合。

除了以上 3 个改进点之外，Grounding DINO 还使用了一种基于语言的预训练方法，叫做 Grounded Pre-Training（GPT），可以在大规模无标注数据上对模型进行预训练，从而提高模型对语言和视觉模态之间关系的理解能力。GPT 主要包括 2 个阶段：

自监督预训练（Self-Supervised Pre-Training）：这个阶段使用了一种基于对比学习的方法，叫做 MOCO，可以在无标注的图像上学习图像特征的表示。同时，这个阶段还使用了一种基于掩码语言模型的方法，叫做 BERT，可以在无标注的文本上学习语言特征的表示。这两种方法分别对应于图像编码器和语言编码器，可以分别提取图像和语言的低层特征，并且可以通过一个对齐损失函数来进行联合优化，从而使得图像和语言的特征在同一个空间中对齐。
监督预训练（Supervised Pre-Training）：这个阶段使用了一种基于多任务学习的方法，可以在有标注的数据上对模型进行微调，从而提高模型对语言和视觉模态之间关系的理解能力。这个阶段主要包括 3 个任务：
- 目标检测（Object Detection）：这个任务使用 COCO 数据集，是一个常用的目标检测数据集，包含了 80 个类别和超过 20 万张图像。这个任务可以使模型学习如何根据类别名字来检测出图像中的目标，并给出其位置和类别信息。
- 属性检测（Attribute Detection）：这个任务使用 LVIS 数据集，是一个新颖的目标检测数据集，包含了 1200 个类别和超过 100 万张图像。这个任务可以使模型学习如何根据属性描述来检测出图像中的目标，并给出其位置和属性信息。
- 指代表达理解（Referring Expression Comprehension）：这个任务使用 RefCOCO/+/g 数据集，是一个常用的指代表达理解数据集，包含了超过 14 万个指代表达和超过 5 万张图像。这个任务可以使模型学习如何根据指代表达来检测出图像中的目标，并且给出其位置和指代信息。

通过以上 2 个阶段的预训练，模型可以在不同的语言输入形式下，对不同的目标类别进行有效的检测。

SD的启动命令：

bash 复制代码

nohup python -u launch.py --listen --port 9301 --xformers --no-half-vae --enable-insecure-extension-access --theme dark --gradio-queue > nohup.62.out &

参数 --xformers 有效降低显存占用，提升出图速度。

示例图像，提示词来源于真实图像的导出：

bash 复制代码

1girl,bag,black hair,earrings,full body,glasses,handbag,jewelry,lipstick,looking at viewer,makeup,pantyhose,pencil skirt,polka dot,polka dot dress,sheer legwear,skirt,smile,solo,standing,sunglasses,wall,
a woman in a pink dress leaning against a wall with a white purse and a handbag on her hip,Carol Bove,feminine,a cubist painting,op art,
best quality,masterpiece,ultra high res,(photorealistic:1.4),
Negative prompt: (badhandv4:1.2),(ng_deepnegative_v1_75t, bad_prompt_version2-neg, EasyNegative:0.9),
(worst quality, low quality:1.3),(depth of field, blurry:1.2),(greyscale, monochrome:1.1),croped,lowres,text,jpeg artifacts,(logo,signature,watermark,username,artist name,title:1.3),
Steps: 30, Sampler: DPM++ 2M SDE Karras, CFG scale: 7, Seed: 3576157745, Size: 768x1024, Model hash: e4a30e4607, Model: 麦橘写实_MajicMIX_Realistic_v6, Denoising strength: 0.3, Clip skip: 2, ADetailer model: face_yolov8n.pt, ADetailer prompt: "detailed face, close-up, portrait,", ADetailer confidence: 0.3, ADetailer dilate/erode: 4, ADetailer mask blur: 4, ADetailer denoising strength: 0.4, ADetailer inpaint only masked: True, ADetailer inpaint padding: 32, ADetailer model 2nd: hand_yolov8s.pt, ADetailer prompt 2nd: "detailed hand, perfect hand,", ADetailer confidence 2nd: 0.3, ADetailer dilate/erode 2nd: 4, ADetailer mask blur 2nd: 4, ADetailer denoising strength 2nd: 0.4, ADetailer inpaint only masked 2nd: True, ADetailer inpaint padding 2nd: 32, ADetailer version: 23.7.6, Hires upscale: 2, Hires steps: 10, Hires upscaler: 8x_NMKD-Superscale_150000_G, Version: v1.4.0
Used embeddings: badhandv4 [dba1], ng_deepnegative_v1_75t [1a3e], bad_prompt_version2-neg [afea], EasyNegative [119b]

图像：

SAM 插件：segment-anything

GroundingDINO的Huggingface工程：Huggingface - GroundingDINO

1. SAM 模型

安装插件，搜索segment anything，下载 SAM 模型，即：

cd stable_diffusion_webui_docker/extensions/sd-webui-segment-anything/models/sam

bypy downfile /stable_diffusion/extensions/segment_anything/sam_vit_h_4b8939.pth sam_vit_h_4b8939.pth

启用 Segment Anything (分离图像元素) 脚本，点击 人物的裙子，出现黑点，即：

再点击 预览分割结果，即出现全部的分割项，物体从小到大，预览有些形变，真实图像正常，即：

即将所需的裙子部分全部分割出来。

2. GroundingDINO 模型

进入 segment anything 的配置页面，启用本地 Groudingdino 功能，这样就可以直接使用 Huggingface 的工程：

下载 Huggingface 的 GroundingDINO 组件：

cd stable_diffusion_webui_docker/extensions/sd-webui-segment-anything/models/

bypy downdir /huggingface/GroundingDINO grounding-dino

在使用时，还需下载 bert-base-uncased 与 tokenizer.json，大约 440 M。

启用 GroundingDINO：

选择所使用的模型，主要用于目标检测。
选择提示词，用于标识所要分割物体。

即：

即 GroundingDINO 通过提示词，进行目标检测，再结合 Segment Anything 的实例分割，可以取得更好的性能。

Bug: 遇到 [Errno 39] Directory not empty: '/stable_diffusion_webui_docker/venv/lib/python3.8/site-packages/~IL'

关闭 SD 服务，删除 ~IL 文件夹，再重新启动，即可。

3. 局部重绘

保持之前的图像与 Mask，原图与模版的效果如下：

使用 图生图 的 上传重绘模版 功能，上传原图与Mask，设置如下：

缩放模式：仅调整大小
蒙版模式：重绘蒙版内容
蒙版区域内容处理：填充(fill)
重绘区域：整张图片 ，边缘更加和谐
同时开启 ControlNet 的 软边缘(SoftEdge) 模式，强化纹理效果。

即：

提示词：

yellow pencil skirt,polka dot dress,skirt,

yellow pencil skirt,polka dot dress,skirt,

best quality,masterpiece,ultra high res,(photorealistic:1.4),

负向提示词：

(ng_deepnegative_v1_75t, bad_prompt_version2-neg, EasyNegative:0.9),

(worst quality, low quality:1.3),(depth of field, blurry:1.2),(greyscale, monochrome:1.1),lowres,jpeg artifacts,

增加 局部重绘 (Inpaint) 功能，同时，控制模式 选择 更偏向提示词。

裙子颜色的依次效果如下：

袜子提示词，修改颜色属性，注意不同随机种子差别较大：

(black color wrap hip pantyhose:1.3),thights,stockings,high heels,

best quality,masterpiece,ultra high res,(photorealistic:1.4),

采样模式：DDIM，随机种子：3474825489，ControlNet控制模型：更偏向提示词，提升重绘幅度至：0.8

颜色：image、black、pink、blue、yellow，袜子颜色的依次效果如下：

具体信息：

(black color wrap hip pantyhose:1.3),thights,stockings,high heels,

best quality,masterpiece,ultra high res,(photorealistic:1.4),

Negative prompt: (ng_deepnegative_v1_75t, bad_prompt_version2-neg, EasyNegative:0.9),

(worst quality, low quality:1.3),(depth of field, blurry:1.2),(greyscale, monochrome:1.1),lowres,jpeg artifacts,

Steps: 20, Sampler: DDIM, CFG scale: 7, Seed: 3474825489, Size: 1536x2048, Model hash: e4a30e4607, Model: 麦橘写实_MajicMIX_Realistic_v6, Denoising strength: 0.8, Clip skip: 2, Mask blur: 4, ControlNet 0: "preprocessor: none, model: control_v11p_sd15_softedge [a8575a2a], weight: 1, starting/ending: (0, 1), resize mode: Just Resize, pixel perfect: True, control mode: My prompt is more important, preprocessor params: (512, -1, -1)", Version: v1.4.0

即通过 Segment Anything + Grounding DINO + ControlNet + Inpaint 实现局部重绘。

Stable Diffusion - 扩展 SegmentAnything 和 GroundingDINO 实例分割算法 插件的配置与使用

1. SAM 模型

2. GroundingDINO 模型

3. 局部重绘

Stable Diffusion - 扩展 SegmentAnything 和 GroundingDINO 实例分割算法插件的配置与使用