Grounded-Segment-Anything 项目教程

毛宝锋

于 2024-08-08 08:05:56 发布

阅读量959

点赞数 16

本文链接：https://blog.csdn.net/gitblog_00752/article/details/141013109

版权

Grounded-Segment-Anything 项目教程

Grounded-Segment-AnythingGrounded-SAM: Marrying Grounding-DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything项目地址:https://gitcode.com/gh_mirrors/gr/Grounded-Segment-Anything

项目介绍

Grounded-Segment-Anything 是一个结合了 Grounding DINO 和 Segment Anything 的开源项目，旨在通过文本输入实现对象的检测和分割。该项目由 IDEA-Research 开发，支持多种功能，包括自动分割、检测、交互式分割和图像修复等。

项目快速启动

环境配置

首先，确保你已经安装了必要的依赖包：

pip install -r requirements.txt

下载预训练模型

下载 Grounding DINO 的预训练权重：

cd Grounded-Segment-Anything
# 下载预训练的 groundingdino-swin-tiny 模型
wget https://path-to-pretrained-model/groundingdino-swin-tiny.pth

运行示例

以下是一个简单的示例代码，展示如何使用 Grounded-Segment-Anything 进行图像分割：

import groundingdino
from segment_anything import sam_model_registry, SamPredictor

# 加载模型
grounding_dino_model = groundingdino.load_model('groundingdino-swin-tiny.pth')
sam_model = sam_model_registry['default'](checkpoint='path-to-sam-checkpoint')

# 创建预测器
predictor = SamPredictor(sam_model)

# 加载图像
image = cv2.imread('path-to-image.jpg')

# 进行分割
masks, _, _ = predictor.predict(
    point_coords=None,
    point_labels=None,
    box=None,
    mask_input=None,
    multimask_output=False,
)

# 显示结果
cv2.imshow('Segmentation Result', masks[0])
cv2.waitKey(0)
cv2.destroyAllWindows()

应用案例和最佳实践

自动分割

自动分割功能可以一次性分割整个图像，无需指定任何提示：

# 自动分割示例
automask = groundingdino.auto_mask(image)
cv2.imshow('Auto Mask', automask)
cv2.waitKey(0)
cv2.destroyAllWindows()

文本交互式分割

结合 Grounding DINO 和 Segment Anything 实现文本交互式分割：

# 文本交互式分割示例
text_prompt = "cat"
seg_result = groundingdino.seg(image, text_prompt)
cv2.imshow('Segmentation Result', seg_result)
cv2.waitKey(0)
cv2.destroyAllWindows()