工具分享

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

mobilesam-logo

特点

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

部署

  • 安装项目

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

  • 使用

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    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


声明

本文仅作为工具分享记录。

本文永久记录于区块链博客xlog。