工具分享

​ MobileSAM比FastSAM更小，速度更快，并且与SAM的处理流程更接近。

特点

MobileSAM 比当前的 FastSAM 小约 7 倍，快约 5 倍，特点就是又快又好。

部署

• 安装项目

  git clone [email protected]:ChaoningZhang/MobileSAM.git && cd MobileSAM
  pip install -e .
  

• 使用

  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保持同步。