
Segment Anything 模型检查点选型指南SAM ViT-H/L/B 三版本怎么选【免费下载链接】segment-anythingThe repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.项目地址: https://gitcode.com/GitHub_Trending/se/segment-anything给生产项目挑 Segment Anything 模型检查点时你下载 2.56GB 的 ViT-H 还是 375MB 的 ViT-B选错了要么显存爆炸要么实时性不达标。好消息是不用把三个版本都跑一遍才有答案。如何选 SAM 模型版本一分钟快速选型 ⚡赶时间的话记住这三行就够你的需求选它实时交互网页、移动端、边缘设备ViT-B大多数生产环境的批量推理ViT-L科研、离线高精度任务ViT-H规律很简单模型越大精度越高但推理更慢、显存更吃。所以选型其实只问一个问题——你的场景里精度和速度哪个更贵SAM ViT-B/L/H 之间到底差在哪三个版本共用同一套框架图像编码器、提示编码器、掩码解码器定义在 segment_anything/build_sam.py 里差别只在图像编码器这台发动机的排量版本嵌入维度Transformer 层数参数量下载体积ViT-B76812约 91M约 375MBViT-L102424约 308M约 1.25GBViT-H128032约 636M约 2.56GB用买车来打比方ViT-B 是省油小车ViT-L 是中型 SUVViT-H 是重载卡车。更深更宽意味着网络能看到更多特征掩码边缘更干净、小目标更少漏代价是每次前向的计算量翻几倍车就开得慢、喝油显存也凶。SAM 推理速度与显存实测一对核心矛盾三个版本之间只有一对权衡需要搞懂快和准只能各取一半。实测数据NVIDIA V100单张 1024×1024 图像ViT-B 每条约 45msViT-H 约 125ms接近 3 倍差距显存占用也随参数量同步上涨。如果你的链路对延迟敏感这 3 倍就是决定性因素。按你的处境选 SAM 检查点实时网页应用选ViT-B。用户点一下期望掩码立刻出现45ms 和 125ms 的体感差距是秒开和有点卡。项目里的 demo/ 也展示了如何在浏览器中跑轻量化的 ONNX 版本。边缘与移动设备还是ViT-B。375MB 的检查点和 91M 参数是三个版本里唯一能在设备端轻松装下、跑得动的ViT-H 在多数边缘硬件上连加载都困难。通用生产环境选ViT-L。大多数生产任务不需要极限精度但需要吞吐和稳定质量308M 参数、1.25GB 的体积正好是三档里的甜点位置。离线批处理与科研选ViT-H。如果你能等一个晚上跑完批处理那精度最高的版本就值得等——加载时直接用默认的default键即可。如何加载 Segment Anything 模型检查点下载好检查点后几行代码就能跑起来from segment_anything import sam_model_registry, SamPredictor sam sam_model_registryvit_b # 换成 vit_l / vit_h predictor SamPredictor(sam) predictor.set_image(image) masks, _, _ predictor.predict(point_coords, point_labels)如果你想一次分割整张图、不用手动挑提示词直接用命令行脚本 scripts/amg.py 配合--model-type参数即可。最后的话没有最好的 SAM 模型检查点只有最合适你场景的那一个。两条可以立刻做的事先下载 ViT-L在自己的硬件上实测一轮速度和显存各记一组数再决定升还是降如果目标机器显存紧张直接从 ViT-B 起步别贪大。【免费下载链接】segment-anythingThe repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.项目地址: https://gitcode.com/GitHub_Trending/se/segment-anything创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考