Sam2 Hiera Large

🚀 SAM2-Hiera-large

本倉庫包含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-large", filename="sam2_hiera_large.pt", local_dir = "./")
sam2_checkpoint = "../checkpoints/sam2_hiera_large.pt"
model_cfg = "sam2_hiera_l.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-large", filename="sam2_hiera_large.pt", local_dir = "./")
sam2_checkpoint = "../checkpoints/sam2_hiera_large.pt"
model_cfg = "sam2_hiera_l.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許可證。