File size: 45,007 Bytes
fe4d363 013cfc4 fe4d363 77e71bc fe4d363 6420584 77e71bc fe4d363 8964976 8666c77 fe4d363 5fe4acc fe4d363 5fe4acc 8964976 8666c77 a04831d 8964976 a04831d 8666c77 a04831d 8666c77 a04831d 8666c77 a04831d 8666c77 a04831d 8666c77 a04831d 8666c77 5fe4acc fe4d363 5fe4acc fe4d363 8c93b19 fe4d363 8c93b19 a04831d 7fbec75 fe4d363 a04831d fe4d363 a04831d fe4d363 a04831d fe4d363 a04831d fe4d363 a04831d 7fbec75 a04831d 7fbec75 a04831d 7fbec75 a04831d 7fbec75 a04831d 7fbec75 a04831d fe4d363 a04831d 7a63b8a a04831d 7a63b8a fe4d363 a04831d 7a63b8a a04831d 7a63b8a a04831d 7a63b8a a04831d 7a63b8a a04831d fe4d363 013cfc4 fe4d363 5fe4acc fe4d363 8964976 fe4d363 8964976 fe4d363 8964976 fe4d363 2a52266 fe4d363 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 | ---
library_name: pytorch
license: apache-2.0
pipeline_tag: image-segmentation
tags:
- foundation
- android
---

# Segment-Anything-Model: Optimized for Mobile Deployment
## High-quality segmentation mask generation around any object in an image with simple input prompt
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.
This model is an implementation of Segment-Anything-Model found [here](https://github.com/facebookresearch/segment-anything).
This repository provides scripts to run Segment-Anything-Model on Qualcomm® devices.
More details on model performance across various devices, can be found
[here](https://aihub.qualcomm.com/models/sam).
### Model Details
- **Model Type:** Semantic segmentation
- **Model Stats:**
- Model checkpoint: vit_l
- Input resolution: 720p (720x1280)
- Number of parameters (SAMDecoder): 5.11M
- Model size (SAMDecoder): 19.6 MB
| Model | Device | Chipset | Target Runtime | Inference Time (ms) | Peak Memory Range (MB) | Precision | Primary Compute Unit | Target Model
|---|---|---|---|---|---|---|---|---|
| SAMDecoder | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 7.496 ms | 0 - 29 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 7.297 ms | 4 - 26 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.so) |
| SAMDecoder | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 10.997 ms | 1 - 59 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.onnx) |
| SAMDecoder | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | TFLITE | 5.194 ms | 0 - 39 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 5.138 ms | 4 - 45 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.so) |
| SAMDecoder | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 7.908 ms | 4 - 58 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.onnx) |
| SAMDecoder | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 5.022 ms | 0 - 35 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 4.808 ms | 4 - 43 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 8.33 ms | 3 - 48 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.onnx) |
| SAMDecoder | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 7.482 ms | 0 - 34 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 6.833 ms | 4 - 7 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | SA7255P ADP | SA7255P | TFLITE | 53.072 ms | 0 - 31 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | SA7255P ADP | SA7255P | QNN | 49.854 ms | 2 - 9 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 7.488 ms | 0 - 33 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | SA8255 (Proxy) | SA8255P Proxy | QNN | 6.856 ms | 4 - 6 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | SA8295P ADP | SA8295P | TFLITE | 10.448 ms | 0 - 35 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | SA8295P ADP | SA8295P | QNN | 9.023 ms | 0 - 14 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 7.468 ms | 0 - 31 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | SA8650 (Proxy) | SA8650P Proxy | QNN | 6.898 ms | 4 - 7 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | SA8775P ADP | SA8775P | TFLITE | 10.481 ms | 0 - 32 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | SA8775P ADP | SA8775P | QNN | 9.672 ms | 1 - 11 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | QCS8450 (Proxy) | QCS8450 Proxy | TFLITE | 8.944 ms | 0 - 39 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.tflite) |
| SAMDecoder | QCS8450 (Proxy) | QCS8450 Proxy | QNN | 8.3 ms | 4 - 47 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 7.387 ms | 4 - 4 MB | FP16 | NPU | Use Export Script |
| SAMDecoder | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 14.654 ms | 13 - 13 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMDecoder.onnx) |
| SAMEncoderPart1 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 209.019 ms | 12 - 69 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 205.936 ms | 12 - 93 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.so) |
| SAMEncoderPart1 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 170.285 ms | 25 - 176 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.onnx) |
| SAMEncoderPart1 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | TFLITE | 148.733 ms | 10 - 653 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 146.052 ms | 12 - 652 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.so) |
| SAMEncoderPart1 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 120.931 ms | 24 - 696 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.onnx) |
| SAMEncoderPart1 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 146.901 ms | 12 - 670 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 142.17 ms | 3 - 654 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 108.126 ms | 24 - 673 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.onnx) |
| SAMEncoderPart1 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 209.39 ms | 12 - 78 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 175.952 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | SA7255P ADP | SA7255P | TFLITE | 1176.456 ms | 12 - 655 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | SA7255P ADP | SA7255P | QNN | 1101.014 ms | 4 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 208.032 ms | 12 - 74 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | SA8255 (Proxy) | SA8255P Proxy | QNN | 178.079 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | SA8295P ADP | SA8295P | TFLITE | 243.729 ms | 12 - 640 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | SA8295P ADP | SA8295P | QNN | 207.581 ms | 0 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 206.899 ms | 12 - 79 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | SA8650 (Proxy) | SA8650P Proxy | QNN | 178.981 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | SA8775P ADP | SA8775P | TFLITE | 252.41 ms | 12 - 655 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | SA8775P ADP | SA8775P | QNN | 211.388 ms | 1 - 11 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | QCS8450 (Proxy) | QCS8450 Proxy | TFLITE | 232.157 ms | 12 - 995 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.tflite) |
| SAMEncoderPart1 | QCS8450 (Proxy) | QCS8450 Proxy | QNN | 222.335 ms | 4 - 965 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 171.549 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart1 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 181.095 ms | 38 - 38 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart1.onnx) |
| SAMEncoderPart2 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 654.976 ms | 12 - 110 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 832.207 ms | 12 - 110 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.so) |
| SAMEncoderPart2 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 694.809 ms | 0 - 201 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.onnx) |
| SAMEncoderPart2 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 661.317 ms | 12 - 1110 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.so) |
| SAMEncoderPart2 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 472.889 ms | 12 - 1148 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 528.224 ms | 12 - 1115 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 462.973 ms | 25 - 1409 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.onnx) |
| SAMEncoderPart2 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 654.376 ms | 12 - 104 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 728.462 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 684.138 ms | 12 - 104 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | SA8255 (Proxy) | SA8255P Proxy | QNN | 744.998 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | SA8295P ADP | SA8295P | TFLITE | 704.249 ms | 12 - 1176 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | SA8295P ADP | SA8295P | QNN | 782.37 ms | 0 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 624.867 ms | 12 - 110 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | SA8650 (Proxy) | SA8650P Proxy | QNN | 743.875 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | SA8775P ADP | SA8775P | TFLITE | 722.805 ms | 0 - 1144 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.tflite) |
| SAMEncoderPart2 | SA8775P ADP | SA8775P | QNN | 739.926 ms | 1 - 11 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 635.324 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart2 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 741.229 ms | 52 - 52 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart2.onnx) |
| SAMEncoderPart3 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 659.742 ms | 12 - 114 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 849.715 ms | 1 - 107 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.so) |
| SAMEncoderPart3 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 687.45 ms | 12 - 211 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.onnx) |
| SAMEncoderPart3 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 682.052 ms | 12 - 1108 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.so) |
| SAMEncoderPart3 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 609.933 ms | 24 - 1425 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.onnx) |
| SAMEncoderPart3 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 423.957 ms | 12 - 1149 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 577.926 ms | 12 - 1114 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 463.85 ms | 24 - 1409 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.onnx) |
| SAMEncoderPart3 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 658.397 ms | 12 - 105 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 719.036 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | SA7255P ADP | SA7255P | QNN | 1877.658 ms | 4 - 13 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 658.984 ms | 12 - 106 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | SA8255 (Proxy) | SA8255P Proxy | QNN | 746.128 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | SA8295P ADP | SA8295P | TFLITE | 707.389 ms | 11 - 1169 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | SA8295P ADP | SA8295P | QNN | 780.808 ms | 0 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 650.209 ms | 12 - 104 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | SA8650 (Proxy) | SA8650P Proxy | QNN | 741.658 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | SA8775P ADP | SA8775P | TFLITE | 699.588 ms | 0 - 1145 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.tflite) |
| SAMEncoderPart3 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 682.704 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart3 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 741.666 ms | 53 - 53 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart3.onnx) |
| SAMEncoderPart4 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 660.425 ms | 12 - 103 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 850.796 ms | 12 - 111 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.so) |
| SAMEncoderPart4 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 735.756 ms | 24 - 217 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.onnx) |
| SAMEncoderPart4 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 676.625 ms | 3 - 1105 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.so) |
| SAMEncoderPart4 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 606.957 ms | 23 - 1418 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.onnx) |
| SAMEncoderPart4 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 469.821 ms | 11 - 1144 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 574.694 ms | 10 - 1118 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 462.452 ms | 36 - 1417 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.onnx) |
| SAMEncoderPart4 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 661.229 ms | 12 - 103 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 727.569 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | SA7255P ADP | SA7255P | QNN | 1869.043 ms | 8 - 17 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 662.568 ms | 12 - 103 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | SA8255 (Proxy) | SA8255P Proxy | QNN | 733.293 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | SA8295P ADP | SA8295P | TFLITE | 706.266 ms | 12 - 1172 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | SA8295P ADP | SA8295P | QNN | 784.772 ms | 0 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 656.939 ms | 12 - 108 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | SA8650 (Proxy) | SA8650P Proxy | QNN | 741.153 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | SA8775P ADP | SA8775P | TFLITE | 717.416 ms | 0 - 1143 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.tflite) |
| SAMEncoderPart4 | SA8775P ADP | SA8775P | QNN | 739.764 ms | 6 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 629.135 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart4 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 742.285 ms | 52 - 52 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart4.onnx) |
| SAMEncoderPart5 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 664.891 ms | 12 - 102 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 818.349 ms | 12 - 120 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.so) |
| SAMEncoderPart5 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 752.745 ms | 24 - 211 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.onnx) |
| SAMEncoderPart5 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 666.563 ms | 6 - 1101 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.so) |
| SAMEncoderPart5 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 578.332 ms | 18 - 1423 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.onnx) |
| SAMEncoderPart5 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 471.325 ms | 12 - 1147 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 577.197 ms | 12 - 1114 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 461.817 ms | 36 - 1424 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.onnx) |
| SAMEncoderPart5 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 655.75 ms | 12 - 118 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 749.162 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | SA7255P ADP | SA7255P | QNN | 1874.378 ms | 4 - 13 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 665.3 ms | 12 - 113 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | SA8255 (Proxy) | SA8255P Proxy | QNN | 736.349 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | SA8295P ADP | SA8295P | TFLITE | 1.844674407370935e+16 ms | 12 - 1172 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | SA8295P ADP | SA8295P | QNN | 783.84 ms | 1 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 654.117 ms | 12 - 115 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | SA8650 (Proxy) | SA8650P Proxy | QNN | 743.603 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | SA8775P ADP | SA8775P | TFLITE | 718.325 ms | 0 - 1144 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.tflite) |
| SAMEncoderPart5 | SA8775P ADP | SA8775P | QNN | 740.649 ms | 12 - 22 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 633.494 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart5 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 742.408 ms | 52 - 52 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart5.onnx) |
| SAMEncoderPart6 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | TFLITE | 655.189 ms | 12 - 108 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | QNN | 807.274 ms | 12 - 120 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.so) |
| SAMEncoderPart6 | Samsung Galaxy S23 | Snapdragon® 8 Gen 2 | ONNX | 755.205 ms | 12 - 204 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.onnx) |
| SAMEncoderPart6 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | QNN | 668.409 ms | 1212 - 2314 MB | FP16 | NPU | [Segment-Anything-Model.so](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.so) |
| SAMEncoderPart6 | Samsung Galaxy S24 | Snapdragon® 8 Gen 3 | ONNX | 584.922 ms | 24 - 1421 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.onnx) |
| SAMEncoderPart6 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | TFLITE | 475.467 ms | 11 - 1140 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | QNN | 528.924 ms | 10 - 1114 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | Snapdragon 8 Elite QRD | Snapdragon® 8 Elite | ONNX | 439.119 ms | 36 - 1413 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.onnx) |
| SAMEncoderPart6 | QCS8550 (Proxy) | QCS8550 Proxy | TFLITE | 665.633 ms | 12 - 108 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | QCS8550 (Proxy) | QCS8550 Proxy | QNN | 727.32 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | SA8255 (Proxy) | SA8255P Proxy | TFLITE | 667.805 ms | 12 - 107 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | SA8255 (Proxy) | SA8255P Proxy | QNN | 742.425 ms | 12 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | SA8295P ADP | SA8295P | TFLITE | 706.978 ms | 12 - 1175 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | SA8295P ADP | SA8295P | QNN | 784.131 ms | 0 - 14 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | SA8650 (Proxy) | SA8650P Proxy | TFLITE | 675.893 ms | 12 - 115 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | SA8650 (Proxy) | SA8650P Proxy | QNN | 739.078 ms | 12 - 15 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | SA8775P ADP | SA8775P | TFLITE | 703.126 ms | 0 - 1144 MB | FP16 | NPU | [Segment-Anything-Model.tflite](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.tflite) |
| SAMEncoderPart6 | SA8775P ADP | SA8775P | QNN | 740.372 ms | 1 - 11 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | Snapdragon X Elite CRD | Snapdragon® X Elite | QNN | 634.143 ms | 12 - 12 MB | FP16 | NPU | Use Export Script |
| SAMEncoderPart6 | Snapdragon X Elite CRD | Snapdragon® X Elite | ONNX | 726.741 ms | 51 - 51 MB | FP16 | NPU | [Segment-Anything-Model.onnx](https://huggingface.co/qualcomm/Segment-Anything-Model/blob/main/SAMEncoderPart6.onnx) |
## Installation
This model can be installed as a Python package via pip.
```bash
pip install "qai-hub-models[sam]"
```
## Configure Qualcomm® AI Hub to run this model on a cloud-hosted device
Sign-in to [Qualcomm® AI Hub](https://app.aihub.qualcomm.com/) with your
Qualcomm® ID. Once signed in navigate to `Account -> Settings -> API Token`.
With this API token, you can configure your client to run models on the cloud
hosted devices.
```bash
qai-hub configure --api_token API_TOKEN
```
Navigate to [docs](https://app.aihub.qualcomm.com/docs/) for more information.
## Demo off target
The package contains a simple end-to-end demo that downloads pre-trained
weights and runs this model on a sample input.
```bash
python -m qai_hub_models.models.sam.demo
```
The above demo runs a reference implementation of pre-processing, model
inference, and post processing.
**NOTE**: If you want running in a Jupyter Notebook or Google Colab like
environment, please add the following to your cell (instead of the above).
```
%run -m qai_hub_models.models.sam.demo
```
### Run model on a cloud-hosted device
In addition to the demo, you can also run the model on a cloud-hosted Qualcomm®
device. This script does the following:
* Performance check on-device on a cloud-hosted device
* Downloads compiled assets that can be deployed on-device for Android.
* Accuracy check between PyTorch and on-device outputs.
```bash
python -m qai_hub_models.models.sam.export
```
```
Profiling Results
------------------------------------------------------------
SAMDecoder
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 7.5
Estimated peak memory usage (MB): [0, 29]
Total # Ops : 845
Compute Unit(s) : NPU (845 ops)
------------------------------------------------------------
SAMEncoderPart1
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 209.0
Estimated peak memory usage (MB): [12, 69]
Total # Ops : 585
Compute Unit(s) : NPU (585 ops)
------------------------------------------------------------
SAMEncoderPart2
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 655.0
Estimated peak memory usage (MB): [12, 110]
Total # Ops : 573
Compute Unit(s) : NPU (573 ops)
------------------------------------------------------------
SAMEncoderPart3
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 659.7
Estimated peak memory usage (MB): [12, 114]
Total # Ops : 573
Compute Unit(s) : NPU (573 ops)
------------------------------------------------------------
SAMEncoderPart4
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 660.4
Estimated peak memory usage (MB): [12, 103]
Total # Ops : 573
Compute Unit(s) : NPU (573 ops)
------------------------------------------------------------
SAMEncoderPart5
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 664.9
Estimated peak memory usage (MB): [12, 102]
Total # Ops : 573
Compute Unit(s) : NPU (573 ops)
------------------------------------------------------------
SAMEncoderPart6
Device : Samsung Galaxy S23 (13)
Runtime : TFLITE
Estimated inference time (ms) : 655.2
Estimated peak memory usage (MB): [12, 108]
Total # Ops : 573
Compute Unit(s) : NPU (573 ops)
```
## How does this work?
This [export script](https://aihub.qualcomm.com/models/sam/qai_hub_models/models/Segment-Anything-Model/export.py)
leverages [Qualcomm® AI Hub](https://aihub.qualcomm.com/) to optimize, validate, and deploy this model
on-device. Lets go through each step below in detail:
Step 1: **Compile model for on-device deployment**
To compile a PyTorch model for on-device deployment, we first trace the model
in memory using the `jit.trace` and then call the `submit_compile_job` API.
```python
import torch
import qai_hub as hub
from qai_hub_models.models.sam import Model
# Load the model
model = Model.from_pretrained()
decoder_model = model.decoder
encoder_splits[0]_model = model.encoder_splits[0]
encoder_splits[1]_model = model.encoder_splits[1]
encoder_splits[2]_model = model.encoder_splits[2]
encoder_splits[3]_model = model.encoder_splits[3]
encoder_splits[4]_model = model.encoder_splits[4]
encoder_splits[5]_model = model.encoder_splits[5]
# Device
device = hub.Device("Samsung Galaxy S23")
# Trace model
decoder_input_shape = decoder_model.get_input_spec()
decoder_sample_inputs = decoder_model.sample_inputs()
traced_decoder_model = torch.jit.trace(decoder_model, [torch.tensor(data[0]) for _, data in decoder_sample_inputs.items()])
# Compile model on a specific device
decoder_compile_job = hub.submit_compile_job(
model=traced_decoder_model ,
device=device,
input_specs=decoder_model.get_input_spec(),
)
# Get target model to run on-device
decoder_target_model = decoder_compile_job.get_target_model()
# Trace model
encoder_splits[0]_input_shape = encoder_splits[0]_model.get_input_spec()
encoder_splits[0]_sample_inputs = encoder_splits[0]_model.sample_inputs()
traced_encoder_splits[0]_model = torch.jit.trace(encoder_splits[0]_model, [torch.tensor(data[0]) for _, data in encoder_splits[0]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[0]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[0]_model ,
device=device,
input_specs=encoder_splits[0]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[0]_target_model = encoder_splits[0]_compile_job.get_target_model()
# Trace model
encoder_splits[1]_input_shape = encoder_splits[1]_model.get_input_spec()
encoder_splits[1]_sample_inputs = encoder_splits[1]_model.sample_inputs()
traced_encoder_splits[1]_model = torch.jit.trace(encoder_splits[1]_model, [torch.tensor(data[0]) for _, data in encoder_splits[1]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[1]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[1]_model ,
device=device,
input_specs=encoder_splits[1]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[1]_target_model = encoder_splits[1]_compile_job.get_target_model()
# Trace model
encoder_splits[2]_input_shape = encoder_splits[2]_model.get_input_spec()
encoder_splits[2]_sample_inputs = encoder_splits[2]_model.sample_inputs()
traced_encoder_splits[2]_model = torch.jit.trace(encoder_splits[2]_model, [torch.tensor(data[0]) for _, data in encoder_splits[2]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[2]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[2]_model ,
device=device,
input_specs=encoder_splits[2]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[2]_target_model = encoder_splits[2]_compile_job.get_target_model()
# Trace model
encoder_splits[3]_input_shape = encoder_splits[3]_model.get_input_spec()
encoder_splits[3]_sample_inputs = encoder_splits[3]_model.sample_inputs()
traced_encoder_splits[3]_model = torch.jit.trace(encoder_splits[3]_model, [torch.tensor(data[0]) for _, data in encoder_splits[3]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[3]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[3]_model ,
device=device,
input_specs=encoder_splits[3]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[3]_target_model = encoder_splits[3]_compile_job.get_target_model()
# Trace model
encoder_splits[4]_input_shape = encoder_splits[4]_model.get_input_spec()
encoder_splits[4]_sample_inputs = encoder_splits[4]_model.sample_inputs()
traced_encoder_splits[4]_model = torch.jit.trace(encoder_splits[4]_model, [torch.tensor(data[0]) for _, data in encoder_splits[4]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[4]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[4]_model ,
device=device,
input_specs=encoder_splits[4]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[4]_target_model = encoder_splits[4]_compile_job.get_target_model()
# Trace model
encoder_splits[5]_input_shape = encoder_splits[5]_model.get_input_spec()
encoder_splits[5]_sample_inputs = encoder_splits[5]_model.sample_inputs()
traced_encoder_splits[5]_model = torch.jit.trace(encoder_splits[5]_model, [torch.tensor(data[0]) for _, data in encoder_splits[5]_sample_inputs.items()])
# Compile model on a specific device
encoder_splits[5]_compile_job = hub.submit_compile_job(
model=traced_encoder_splits[5]_model ,
device=device,
input_specs=encoder_splits[5]_model.get_input_spec(),
)
# Get target model to run on-device
encoder_splits[5]_target_model = encoder_splits[5]_compile_job.get_target_model()
```
Step 2: **Performance profiling on cloud-hosted device**
After compiling models from step 1. Models can be profiled model on-device using the
`target_model`. Note that this scripts runs the model on a device automatically
provisioned in the cloud. Once the job is submitted, you can navigate to a
provided job URL to view a variety of on-device performance metrics.
```python
decoder_profile_job = hub.submit_profile_job(
model=decoder_target_model,
device=device,
)
encoder_splits[0]_profile_job = hub.submit_profile_job(
model=encoder_splits[0]_target_model,
device=device,
)
encoder_splits[1]_profile_job = hub.submit_profile_job(
model=encoder_splits[1]_target_model,
device=device,
)
encoder_splits[2]_profile_job = hub.submit_profile_job(
model=encoder_splits[2]_target_model,
device=device,
)
encoder_splits[3]_profile_job = hub.submit_profile_job(
model=encoder_splits[3]_target_model,
device=device,
)
encoder_splits[4]_profile_job = hub.submit_profile_job(
model=encoder_splits[4]_target_model,
device=device,
)
encoder_splits[5]_profile_job = hub.submit_profile_job(
model=encoder_splits[5]_target_model,
device=device,
)
```
Step 3: **Verify on-device accuracy**
To verify the accuracy of the model on-device, you can run on-device inference
on sample input data on the same cloud hosted device.
```python
decoder_input_data = decoder_model.sample_inputs()
decoder_inference_job = hub.submit_inference_job(
model=decoder_target_model,
device=device,
inputs=decoder_input_data,
)
decoder_inference_job.download_output_data()
encoder_splits[0]_input_data = encoder_splits[0]_model.sample_inputs()
encoder_splits[0]_inference_job = hub.submit_inference_job(
model=encoder_splits[0]_target_model,
device=device,
inputs=encoder_splits[0]_input_data,
)
encoder_splits[0]_inference_job.download_output_data()
encoder_splits[1]_input_data = encoder_splits[1]_model.sample_inputs()
encoder_splits[1]_inference_job = hub.submit_inference_job(
model=encoder_splits[1]_target_model,
device=device,
inputs=encoder_splits[1]_input_data,
)
encoder_splits[1]_inference_job.download_output_data()
encoder_splits[2]_input_data = encoder_splits[2]_model.sample_inputs()
encoder_splits[2]_inference_job = hub.submit_inference_job(
model=encoder_splits[2]_target_model,
device=device,
inputs=encoder_splits[2]_input_data,
)
encoder_splits[2]_inference_job.download_output_data()
encoder_splits[3]_input_data = encoder_splits[3]_model.sample_inputs()
encoder_splits[3]_inference_job = hub.submit_inference_job(
model=encoder_splits[3]_target_model,
device=device,
inputs=encoder_splits[3]_input_data,
)
encoder_splits[3]_inference_job.download_output_data()
encoder_splits[4]_input_data = encoder_splits[4]_model.sample_inputs()
encoder_splits[4]_inference_job = hub.submit_inference_job(
model=encoder_splits[4]_target_model,
device=device,
inputs=encoder_splits[4]_input_data,
)
encoder_splits[4]_inference_job.download_output_data()
encoder_splits[5]_input_data = encoder_splits[5]_model.sample_inputs()
encoder_splits[5]_inference_job = hub.submit_inference_job(
model=encoder_splits[5]_target_model,
device=device,
inputs=encoder_splits[5]_input_data,
)
encoder_splits[5]_inference_job.download_output_data()
```
With the output of the model, you can compute like PSNR, relative errors or
spot check the output with expected output.
**Note**: This on-device profiling and inference requires access to Qualcomm®
AI Hub. [Sign up for access](https://myaccount.qualcomm.com/signup).
## Run demo on a cloud-hosted device
You can also run the demo on-device.
```bash
python -m qai_hub_models.models.sam.demo --on-device
```
**NOTE**: If you want running in a Jupyter Notebook or Google Colab like
environment, please add the following to your cell (instead of the above).
```
%run -m qai_hub_models.models.sam.demo -- --on-device
```
## Deploying compiled model to Android
The models can be deployed using multiple runtimes:
- TensorFlow Lite (`.tflite` export): [This
tutorial](https://www.tensorflow.org/lite/android/quickstart) provides a
guide to deploy the .tflite model in an Android application.
- QNN (`.so` export ): This [sample
app](https://docs.qualcomm.com/bundle/publicresource/topics/80-63442-50/sample_app.html)
provides instructions on how to use the `.so` shared library in an Android application.
## View on Qualcomm® AI Hub
Get more details on Segment-Anything-Model's performance across various devices [here](https://aihub.qualcomm.com/models/sam).
Explore all available models on [Qualcomm® AI Hub](https://aihub.qualcomm.com/)
## License
* The license for the original implementation of Segment-Anything-Model can be found [here](https://github.com/facebookresearch/segment-anything/blob/main/LICENSE).
* The license for the compiled assets for on-device deployment can be found [here](https://qaihub-public-assets.s3.us-west-2.amazonaws.com/qai-hub-models/Qualcomm+AI+Hub+Proprietary+License.pdf)
## References
* [Segment Anything](https://arxiv.org/abs/2304.02643)
* [Source Model Implementation](https://github.com/facebookresearch/segment-anything)
## Community
* Join [our AI Hub Slack community](https://aihub.qualcomm.com/community/slack) to collaborate, post questions and learn more about on-device AI.
* For questions or feedback please [reach out to us](mailto:ai-hub-support@qti.qualcomm.com).
|