MobileSAM-比FastSAM还要Faster的SAM

工具分享#

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

20-mobilesam-logo

特点#

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

部署#

安装项目

git clone git@github.com: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保持同步。