Sam2 Hiera Small

🚀 SAM2-Hiera-small

本仓库包含SAM2模型的小型变体。SAM2是Meta发布的最先进的掩码生成模型。

🚀 快速开始

你可以按照以下步骤使用该模型。首先，安装打包版本的SAM2。

pip install samv2 huggingface_hub

✨ 主要特性

每个模型推理时需要不同的类。

💻 使用示例

基础用法

以下是使用提示进行推理的示例代码：

from huggingface_hub import hf_hub_download
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor

hf_hub_download(repo_id = "merve/sam2-hiera-small", filename="sam2_hiera_small.pt", local_dir = "./")

ckpt = f"./sam2_hiera_small.pt"
config = "sam2_hiera_s.yaml"

sam2_model = build_sam2(config, ckpt, device="cuda", apply_postprocessing=False)
predictor = SAM2ImagePredictor(sam2_model)

# it accepts coco format
box = [x1, y1, w, h]
predictor.set_image(image)

masks = predictor.predict(box=box,
multimask_output=False)

高级用法

以下是自动掩码生成的示例代码：

from huggingface_hub import hf_hub_download
from sam2.build_sam import build_sam2
from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator

hf_hub_download(repo_id = "merve/sam2-hiera-small", filename="sam2_hiera_small.pt", local_dir = "./")
ckpt = f"./sam2_hiera_small.pt"
config = "sam2_hiera_s.yaml"

sam2 = build_sam2(model_cfg, sam2_checkpoint, device ='cuda', apply_postprocessing=False)

mask_generator = SAM2AutomaticMaskGenerator(sam2)
masks = mask_generator.generate(image)

📚 详细文档

SAM2背后的团队为所有任务制作了示例笔记本。

• 有关提示的完整示例，请参阅图像预测器示例。
• 有关生成所有掩码的示例，请参阅自动掩码生成示例。
• 有关视频对象分割的示例，请参阅视频预测器示例。

📄 许可证

本项目采用Apache 2.0许可证。