词元之母TOK.MOM - 平台充值汇率 1:1 即 1 人民币充值到账 1 美元,支持一个 Key 调用近 600+ 海内外模型,限时特价模型低至 1 折,欢迎上岸!
| 来源 | 内置(默认安装) |
| 路径 | skills/mlops/models/segment-anything |
| 版本 | 1.0.0 |
| 作者 | Orchestra Research |
| 许可证 | MIT |
| 依赖项 | segment-anything, transformers>=4.30.0, torch>=1.7.0 |
| 平台 | linux, macos, windows |
| 标签 | Multimodal, Image Segmentation, Computer Vision, SAM, Zero-Shot |
SAM Architecture:
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
│ Image Encoder │────▶│ Prompt Encoder │────▶│ Mask Decoder │
│ (ViT) │ │ (Points/Boxes) │ │ (Transformer) │
└─────────────────┘ └─────────────────┘ └─────────────────┘
│ │ │
Image Embeddings Prompt Embeddings Masks + IoU
(computed once) (per prompt) predictions| 模型 | 检查点 | 大小 | 速度 | 精度 |
|---|---|---|---|---|
| ViT-H | vit_h | 2.4 GB | 最慢 | 最佳 |
| ViT-L | vit_l | 1.2 GB | 中等 | 良好 |
| ViT-B | vit_b | 375 MB | 最快 | 良好 |
| Prompt | 描述 | 使用场景 |
|---|---|---|
| 点(前景) | 点击对象 | 单对象选择 |
| 点(背景) | 点击对象外部 | 排除区域 |
| 边界框 | 对象周围的矩形 | 较大对象 |
| 先前掩码 | 低分辨率掩码输入 | 迭代精化 |
| 问题 | 解决方案 |
|---|---|
| 内存不足 | 使用 ViT-B 模型,缩小图像尺寸 |
| 推理缓慢 | 使用 ViT-B,减小 points_per_side |
| 掩码质量差 | 尝试不同 prompt,使用框 + 点组合 |
| 边缘伪影 | 使用 stability_score 过滤 |
| 小对象漏检 | 增大 points_per_side |