---
license: mit
---
# ONNX models for MobileSAM

* Original repository: [MobileSAM (Github)](https://github.com/ChaoningZhang/MobileSAM)
* Original weights: [MobileSam (HuggingFace)](https://huggingface.co/dhkim2810/MobileSAM)

## 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.