【亲测免费】 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()

典型生态项目

Narapi-SAM

Narapi-SAM 是一个将 Segment Anything 集成到 Narapi 中的项目，Narapi 是一个用于 SAM 的优秀查看器。

Lang-Segment-Anything

Lang-Segment-Anything 是另一个结合 GroundingDINO 和 Segment-Anything 的优秀演示，支持更多的文本交互功能。

通过这些生态项目，Grounded-Segment-Anything 的社区不断壮大，提供了更多有趣和实用的功能。