hughie

ツールの共有#

MobileSAMは FastSAM よりも小さく、速度が速く、SAM の処理フローにより近いです。

特徴#

MobileSAM は現在の FastSAM よりも約 7 倍小さく、約 5 倍速く、速くて良いという特徴があります。

デプロイ#

• プロジェクトのインストール

  Copy

  git clone git@github.com:ChaoningZhang/MobileSAM.git && cd MobileSAM
  pip install -e .
  

• 使用法

  Copy

  from mobile_sam import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor
  
  model_type = "vit_t"
  sam_checkpoint = "./weights/mobile_sam.pt"
  
  device = "cuda" if torch.cuda.is_available() else "cpu"
  
  mobile_sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
  mobile_sam.to(device=device)
  mobile_sam.eval()
  
  predictor = SamPredictor(mobile_sam)
  predictor.set_image(<your_image>)
  # 入力プロンプト
  masks, _, _ = predictor.predict(<input_prompts>)
  

プラットフォーム#

公式デモ

免責事項#

この記事はツールの共有記録としてのみ提供されています。

この記事はHBlogと同期しています。