Sam2 Hiera Large

🚀 SAM2-Hiera-large

このリポジトリには、SAM2モデルの大規模バリアントが含まれています。SAM2は、Metaによってリリースされた最先端のマスク生成モデルです。

🚀 クイックスタート

SAM2-Hiera-largeモデルを使用するには、まず必要なパッケージをインストールし、モデルを構築して推論を行います。

📦 インストール

まず、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ライセンスの下で公開されています。