ONNX models for MobileSAM
- Original repository: MobileSAM (Github)
- Original weights: MobileSam (HuggingFace)
Models
| Model | Description |
|---|---|
| mobile_sam_image_encoder.onnx | Image encoder created from MobileSAM weights |
| sam_mask_decoder_single.onnx | Mask decoder for SegmentAnything, returns a single mask |
| sam_mask_decoder_multi.onnx | Mask decoder for SegmentAnything, returns 4 masks |
Export
The image encoder was exported with export_image_enoder.py found in this repository.
The mask decoders were exported with export_models.py from the MobileSAM repository.
