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>)
  

平台#

官方 demo

声明#

本文僅作為工具分享記錄。

本文與HBlog保持同步。