返回模型
说明文档
MobileSAM ONNX 模型
- 原始仓库：MobileSAM (Github)
- 原始权重：MobileSam (HuggingFace)
模型
| 模型 | 描述 |
|---|---|
| mobile_sam_image_encoder.onnx | 基于 MobileSAM 权重创建的图像编码器 |
| sam_mask_decoder_single.onnx | SegmentAnything 掩码解码器，返回单个掩码 |
| sam_mask_decoder_multi.onnx | SegmentAnything 掩码解码器，返回 4 个掩码 |
导出
图像编码器使用本仓库中的 export_image_enoder.py 导出。
掩码解码器使用 MobileSAM 仓库中的 export_models.py 导出。
Acly/MobileSAM
作者 Acly
↓ 0
♥ 3
创建时间: 2023-08-03 15:43:37+00:00
更新时间: 2023-08-03 16:01:55+00:00
在 Hugging Face 上查看文件 (8)
.gitattributes
.gitignore
README.md
mobile_sam_encoder_onnx/export_image_encoder.py
mobile_sam_encoder_onnx/onnx_image_encoder.py
mobile_sam_image_encoder.onnx
ONNX
sam_mask_decoder_multi.onnx
ONNX
sam_mask_decoder_single.onnx
ONNX