Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
---
|
| 2 |
library_name: pytorch
|
| 3 |
-
license:
|
| 4 |
tags:
|
| 5 |
- foundation
|
| 6 |
- android
|
|
@@ -11,7 +11,7 @@ pipeline_tag: image-segmentation
|
|
| 11 |

|
| 12 |
|
| 13 |
# MobileSam: Optimized for Mobile Deployment
|
| 14 |
-
##
|
| 15 |
|
| 16 |
|
| 17 |
Transformer based encoder-decoder where prompts specify what to segment in an image thereby allowing segmentation without the need for additional training. The image encoder generates embeddings and the lightweight decoder operates on the embeddings for point and mask based image segmentation.
|
|
@@ -26,73 +26,73 @@ More details on model performance across various devices, can be found
|
|
| 26 |
|
| 27 |
### Model Details
|
| 28 |
|
| 29 |
-
- **Model Type:**
|
| 30 |
- **Model Stats:**
|
| 31 |
- Model checkpoint: vit_t
|
| 32 |
- Input resolution: 720p (720x1280)
|
| 33 |
- Number of parameters (MobileSamDecoder): 3.876M
|
| 34 |
- Model size (MobileSamDecoder): 19.6 MB
|
| 35 |
|
| 36 |
-
| Model | Device | Chipset | Target Runtime | Inference Time (ms) | Peak Memory Range (MB) |
|
| 37 |
|---|---|---|---|---|---|---|---|---|
|
| 38 |
-
| SAMEncoder |
|
| 39 |
-
| SAMEncoder |
|
| 40 |
-
| SAMEncoder |
|
| 41 |
-
| SAMEncoder |
|
| 42 |
-
| SAMEncoder |
|
| 43 |
-
| SAMEncoder |
|
| 44 |
-
| SAMEncoder |
|
| 45 |
-
| SAMEncoder |
|
| 46 |
-
| SAMEncoder |
|
| 47 |
-
| SAMEncoder | SA7255P ADP | SA7255P |
|
| 48 |
-
| SAMEncoder |
|
| 49 |
-
| SAMEncoder | SA8255 (Proxy) | SA8255P Proxy |
|
| 50 |
-
| SAMEncoder |
|
| 51 |
-
| SAMEncoder | SA8295P ADP | SA8295P |
|
| 52 |
-
| SAMEncoder |
|
| 53 |
-
| SAMEncoder | SA8650 (Proxy) | SA8650P Proxy |
|
| 54 |
-
| SAMEncoder |
|
| 55 |
-
| SAMEncoder | SA8775P ADP | SA8775P |
|
| 56 |
-
| SAMEncoder |
|
| 57 |
-
| SAMEncoder |
|
| 58 |
-
| SAMEncoder |
|
| 59 |
-
| SAMEncoder |
|
| 60 |
-
| SAMEncoder |
|
| 61 |
-
| SAMEncoder |
|
| 62 |
-
| SAMEncoder |
|
| 63 |
-
| SAMEncoder |
|
| 64 |
-
| SAMEncoder |
|
| 65 |
-
| SAMEncoder | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN |
|
| 66 |
-
| SAMEncoder | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX |
|
| 67 |
-
| SAMDecoder |
|
| 68 |
-
| SAMDecoder |
|
| 69 |
-
| SAMDecoder |
|
| 70 |
-
| SAMDecoder |
|
| 71 |
-
| SAMDecoder |
|
| 72 |
-
| SAMDecoder |
|
| 73 |
-
| SAMDecoder |
|
| 74 |
-
| SAMDecoder |
|
| 75 |
-
| SAMDecoder |
|
| 76 |
-
| SAMDecoder | SA7255P ADP | SA7255P |
|
| 77 |
-
| SAMDecoder |
|
| 78 |
-
| SAMDecoder | SA8255 (Proxy) | SA8255P Proxy |
|
| 79 |
-
| SAMDecoder |
|
| 80 |
-
| SAMDecoder | SA8295P ADP | SA8295P |
|
| 81 |
-
| SAMDecoder |
|
| 82 |
-
| SAMDecoder | SA8650 (Proxy) | SA8650P Proxy |
|
| 83 |
-
| SAMDecoder |
|
| 84 |
-
| SAMDecoder | SA8775P ADP | SA8775P |
|
| 85 |
-
| SAMDecoder |
|
| 86 |
-
| SAMDecoder |
|
| 87 |
-
| SAMDecoder |
|
| 88 |
-
| SAMDecoder |
|
| 89 |
-
| SAMDecoder |
|
| 90 |
-
| SAMDecoder |
|
| 91 |
-
| SAMDecoder |
|
| 92 |
-
| SAMDecoder |
|
| 93 |
-
| SAMDecoder |
|
| 94 |
-
| SAMDecoder | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 6.
|
| 95 |
-
| SAMDecoder | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 10.
|
| 96 |
|
| 97 |
|
| 98 |
|
|
@@ -154,21 +154,21 @@ python -m qai_hub_models.models.mobilesam.export
|
|
| 154 |
Profiling Results
|
| 155 |
------------------------------------------------------------
|
| 156 |
SAMEncoder
|
| 157 |
-
Device :
|
| 158 |
-
Runtime : TFLITE
|
| 159 |
-
Estimated inference time (ms) :
|
| 160 |
-
Estimated peak memory usage (MB): [
|
| 161 |
-
Total # Ops : 592
|
| 162 |
-
Compute Unit(s) :
|
| 163 |
|
| 164 |
------------------------------------------------------------
|
| 165 |
SAMDecoder
|
| 166 |
-
Device :
|
| 167 |
-
Runtime : TFLITE
|
| 168 |
-
Estimated inference time (ms) :
|
| 169 |
-
Estimated peak memory usage (MB): [0,
|
| 170 |
-
Total # Ops : 845
|
| 171 |
-
Compute Unit(s) :
|
| 172 |
```
|
| 173 |
|
| 174 |
|
|
|
|
| 1 |
---
|
| 2 |
library_name: pytorch
|
| 3 |
+
license: other
|
| 4 |
tags:
|
| 5 |
- foundation
|
| 6 |
- android
|
|
|
|
| 11 |

|
| 12 |
|
| 13 |
# MobileSam: Optimized for Mobile Deployment
|
| 14 |
+
## Faster Segment Anything: Towards lightweight SAM for mobile applications
|
| 15 |
|
| 16 |
|
| 17 |
Transformer based encoder-decoder where prompts specify what to segment in an image thereby allowing segmentation without the need for additional training. The image encoder generates embeddings and the lightweight decoder operates on the embeddings for point and mask based image segmentation.
|
|
|
|
| 26 |
|
| 27 |
### Model Details
|
| 28 |
|
| 29 |
+
- **Model Type:** Model_use_case.semantic_segmentation
|
| 30 |
- **Model Stats:**
|
| 31 |
- Model checkpoint: vit_t
|
| 32 |
- Input resolution: 720p (720x1280)
|
| 33 |
- Number of parameters (MobileSamDecoder): 3.876M
|
| 34 |
- Model size (MobileSamDecoder): 19.6 MB
|
| 35 |
|
| 36 |
+
| Model | Precision | Device | Chipset | Target Runtime | Inference Time (ms) | Peak Memory Range (MB) | Primary Compute Unit | Target Model
|
| 37 |
|---|---|---|---|---|---|---|---|---|
|
| 38 |
+
| SAMEncoder | float | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | TFLITE | 778.225 ms | 33 - 165 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 39 |
+
| SAMEncoder | float | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | QNN | 501.051 ms | 4 - 14 MB | NPU | Use Export Script |
|
| 40 |
+
| SAMEncoder | float | QCS8450 (Proxy) | Qualcomm® QCS8450 (Proxy) | TFLITE | 657.898 ms | 33 - 175 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 41 |
+
| SAMEncoder | float | QCS8450 (Proxy) | Qualcomm® QCS8450 (Proxy) | QNN | 479.564 ms | 12 - 592 MB | NPU | Use Export Script |
|
| 42 |
+
| SAMEncoder | float | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | TFLITE | 409.044 ms | 33 - 58 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 43 |
+
| SAMEncoder | float | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | QNN | 265.351 ms | 12 - 15 MB | NPU | Use Export Script |
|
| 44 |
+
| SAMEncoder | float | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | TFLITE | 420.56 ms | 31 - 164 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 45 |
+
| SAMEncoder | float | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | QNN | 274.861 ms | 1 - 15 MB | NPU | Use Export Script |
|
| 46 |
+
| SAMEncoder | float | SA7255P ADP | Qualcomm® SA7255P | TFLITE | 778.225 ms | 33 - 165 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 47 |
+
| SAMEncoder | float | SA7255P ADP | Qualcomm® SA7255P | QNN | 501.051 ms | 4 - 14 MB | NPU | Use Export Script |
|
| 48 |
+
| SAMEncoder | float | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | TFLITE | 433.345 ms | 33 - 58 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 49 |
+
| SAMEncoder | float | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | QNN | 268.807 ms | 14 - 15 MB | NPU | Use Export Script |
|
| 50 |
+
| SAMEncoder | float | SA8295P ADP | Qualcomm® SA8295P | TFLITE | 585.898 ms | 33 - 168 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 51 |
+
| SAMEncoder | float | SA8295P ADP | Qualcomm® SA8295P | QNN | 425.029 ms | 0 - 18 MB | NPU | Use Export Script |
|
| 52 |
+
| SAMEncoder | float | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | TFLITE | 421.955 ms | 33 - 59 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 53 |
+
| SAMEncoder | float | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | QNN | 268.324 ms | 15 - 17 MB | NPU | Use Export Script |
|
| 54 |
+
| SAMEncoder | float | SA8775P ADP | Qualcomm® SA8775P | TFLITE | 420.56 ms | 31 - 164 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 55 |
+
| SAMEncoder | float | SA8775P ADP | Qualcomm® SA8775P | QNN | 274.861 ms | 1 - 15 MB | NPU | Use Export Script |
|
| 56 |
+
| SAMEncoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | TFLITE | 417.766 ms | 33 - 60 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 57 |
+
| SAMEncoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | QNN | 269.765 ms | 12 - 87 MB | NPU | Use Export Script |
|
| 58 |
+
| SAMEncoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | ONNX | 388.846 ms | 93 - 149 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 59 |
+
| SAMEncoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | TFLITE | 336.863 ms | 33 - 156 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 60 |
+
| SAMEncoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | QNN | 201.573 ms | 12 - 618 MB | NPU | Use Export Script |
|
| 61 |
+
| SAMEncoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | ONNX | 295.691 ms | 94 - 209 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 62 |
+
| SAMEncoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | TFLITE | 274.65 ms | 33 - 166 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 63 |
+
| SAMEncoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | QNN | 174.753 ms | 12 - 596 MB | NPU | Use Export Script |
|
| 64 |
+
| SAMEncoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | ONNX | 255.159 ms | 74 - 204 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 65 |
+
| SAMEncoder | float | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 276.054 ms | 12 - 12 MB | NPU | Use Export Script |
|
| 66 |
+
| SAMEncoder | float | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 431.108 ms | 131 - 131 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 67 |
+
| SAMDecoder | float | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | TFLITE | 16.756 ms | 0 - 40 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 68 |
+
| SAMDecoder | float | QCS8275 (Proxy) | Qualcomm® QCS8275 (Proxy) | QNN | 13.95 ms | 4 - 14 MB | NPU | Use Export Script |
|
| 69 |
+
| SAMDecoder | float | QCS8450 (Proxy) | Qualcomm® QCS8450 (Proxy) | TFLITE | 8.824 ms | 0 - 44 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 70 |
+
| SAMDecoder | float | QCS8450 (Proxy) | Qualcomm® QCS8450 (Proxy) | QNN | 8.649 ms | 4 - 50 MB | NPU | Use Export Script |
|
| 71 |
+
| SAMDecoder | float | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | TFLITE | 7.359 ms | 0 - 26 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 72 |
+
| SAMDecoder | float | QCS8550 (Proxy) | Qualcomm® QCS8550 (Proxy) | QNN | 6.178 ms | 4 - 7 MB | NPU | Use Export Script |
|
| 73 |
+
| SAMDecoder | float | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | TFLITE | 8.639 ms | 0 - 41 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 74 |
+
| SAMDecoder | float | QCS9075 (Proxy) | Qualcomm® QCS9075 (Proxy) | QNN | 7.319 ms | 3 - 17 MB | NPU | Use Export Script |
|
| 75 |
+
| SAMDecoder | float | SA7255P ADP | Qualcomm® SA7255P | TFLITE | 16.756 ms | 0 - 40 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 76 |
+
| SAMDecoder | float | SA7255P ADP | Qualcomm® SA7255P | QNN | 13.95 ms | 4 - 14 MB | NPU | Use Export Script |
|
| 77 |
+
| SAMDecoder | float | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | TFLITE | 7.358 ms | 0 - 28 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 78 |
+
| SAMDecoder | float | SA8255 (Proxy) | Qualcomm® SA8255P (Proxy) | QNN | 6.289 ms | 4 - 6 MB | NPU | Use Export Script |
|
| 79 |
+
| SAMDecoder | float | SA8295P ADP | Qualcomm® SA8295P | TFLITE | 9.889 ms | 0 - 36 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 80 |
+
| SAMDecoder | float | SA8295P ADP | Qualcomm® SA8295P | QNN | 7.431 ms | 0 - 18 MB | NPU | Use Export Script |
|
| 81 |
+
| SAMDecoder | float | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | TFLITE | 7.342 ms | 0 - 28 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 82 |
+
| SAMDecoder | float | SA8650 (Proxy) | Qualcomm® SA8650P (Proxy) | QNN | 6.196 ms | 6 - 8 MB | NPU | Use Export Script |
|
| 83 |
+
| SAMDecoder | float | SA8775P ADP | Qualcomm® SA8775P | TFLITE | 8.639 ms | 0 - 41 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 84 |
+
| SAMDecoder | float | SA8775P ADP | Qualcomm® SA8775P | QNN | 7.319 ms | 3 - 17 MB | NPU | Use Export Script |
|
| 85 |
+
| SAMDecoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | TFLITE | 7.44 ms | 0 - 31 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 86 |
+
| SAMDecoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | QNN | 6.139 ms | 4 - 21 MB | NPU | Use Export Script |
|
| 87 |
+
| SAMDecoder | float | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 Mobile | ONNX | 8.985 ms | 1 - 61 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 88 |
+
| SAMDecoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | TFLITE | 5.251 ms | 0 - 49 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 89 |
+
| SAMDecoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | QNN | 4.201 ms | 4 - 50 MB | NPU | Use Export Script |
|
| 90 |
+
| SAMDecoder | float | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 Mobile | ONNX | 5.977 ms | 4 - 71 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 91 |
+
| SAMDecoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | TFLITE | 5.027 ms | 0 - 42 MB | NPU | [MobileSam.tflite](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.tflite) |
|
| 92 |
+
| SAMDecoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | QNN | 3.132 ms | 4 - 45 MB | NPU | Use Export Script |
|
| 93 |
+
| SAMDecoder | float | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite Mobile | ONNX | 5.397 ms | 0 - 58 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 94 |
+
| SAMDecoder | float | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 6.674 ms | 4 - 4 MB | NPU | Use Export Script |
|
| 95 |
+
| SAMDecoder | float | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 10.051 ms | 11 - 11 MB | NPU | [MobileSam.onnx](https://huggingface.co/qualcomm/MobileSam/blob/main/MobileSam.onnx) |
|
| 96 |
|
| 97 |
|
| 98 |
|
|
|
|
| 154 |
Profiling Results
|
| 155 |
------------------------------------------------------------
|
| 156 |
SAMEncoder
|
| 157 |
+
Device : cs_8275 (ANDROID 14)
|
| 158 |
+
Runtime : TFLITE
|
| 159 |
+
Estimated inference time (ms) : 778.2
|
| 160 |
+
Estimated peak memory usage (MB): [33, 165]
|
| 161 |
+
Total # Ops : 592
|
| 162 |
+
Compute Unit(s) : npu (532 ops) gpu (0 ops) cpu (60 ops)
|
| 163 |
|
| 164 |
------------------------------------------------------------
|
| 165 |
SAMDecoder
|
| 166 |
+
Device : cs_8275 (ANDROID 14)
|
| 167 |
+
Runtime : TFLITE
|
| 168 |
+
Estimated inference time (ms) : 16.8
|
| 169 |
+
Estimated peak memory usage (MB): [0, 40]
|
| 170 |
+
Total # Ops : 845
|
| 171 |
+
Compute Unit(s) : npu (845 ops) gpu (0 ops) cpu (0 ops)
|
| 172 |
```
|
| 173 |
|
| 174 |
|