| 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_image_enoder.py found in this repository.
The mask decoders were exported with export_models.py from the MobileSAM repository.