Segformer-Base: Optimized for Qualcomm Devices
Segformer Base is a machine learning model that predicts masks and classes of objects in an image.
This is based on the implementation of Segformer-Base found here. This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the Qualcomm® AI Hub Models library to export with custom configurations. More details on model performance across various devices, can be found here.
Qualcomm AI Hub Models uses Qualcomm AI Hub Workbench to compile, profile, and evaluate this model. Sign up to run these models on a hosted Qualcomm® device.
Getting Started
There are two ways to deploy this model on your device:
Option 1: Download Pre-Exported Models
Below are pre-exported model assets ready for deployment.
| Runtime | Precision | Chipset | SDK Versions | Download |
|---|---|---|---|---|
| ONNX | float | Universal | QAIRT 2.45, ONNX Runtime 1.25.0 | Download |
| ONNX | w8a16 | Universal | QAIRT 2.45, ONNX Runtime 1.25.0 | Download |
| ONNX | w8a8 | Universal | QAIRT 2.45, ONNX Runtime 1.25.0 | Download |
| QNN_DLC | float | Universal | QAIRT 2.45 | Download |
| QNN_DLC | w8a16 | Universal | QAIRT 2.45 | Download |
| QNN_DLC | w8a8 | Universal | QAIRT 2.45 | Download |
| TFLITE | float | Universal | QAIRT 2.45 | Download |
| TFLITE | w8a8 | Universal | QAIRT 2.45 | Download |
For more device-specific assets and performance metrics, visit Segformer-Base on Qualcomm® AI Hub.
Option 2: Export with Custom Configurations
Use the Qualcomm® AI Hub Models Python library to compile and export the model with your own:
- Custom weights (e.g., fine-tuned checkpoints)
- Custom input shapes
- Target device and runtime configurations
This option is ideal if you need to customize the model beyond the default configuration provided here.
See our repository for Segformer-Base on GitHub for usage instructions.
Model Details
Model Type: Model_use_case.semantic_segmentation
Model Stats:
- Model checkpoint: nvidia/segformer-b0-finetuned-ade-512-512
- Input resolution: 512x512
- Number of output classes: 150
- Number of parameters: 3.75M
- Model size (float): 14.4 MB
- Model size (w8a16): 4.57 MB
- Model size (w8a8): 3.90 MB
Performance Summary
| Model | Runtime | Precision | Chipset | Inference Time (ms) | Peak Memory Range (MB) | Primary Compute Unit |
|---|---|---|---|---|---|---|
| Segformer-Base | ONNX | float | Snapdragon® X2 Elite | 13.523 ms | 35 - 35 MB | NPU |
| Segformer-Base | ONNX | float | Snapdragon® X Elite | 24.584 ms | 35 - 35 MB | NPU |
| Segformer-Base | ONNX | float | Snapdragon® 8 Gen 3 Mobile | 17.358 ms | 25 - 241 MB | NPU |
| Segformer-Base | ONNX | float | Snapdragon® 8 Gen 1 Mobile | 24.781 ms | 13 - 232 MB | NPU |
| Segformer-Base | ONNX | float | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 23.739 ms | 22 - 149 MB | NPU |
| Segformer-Base | ONNX | float | Qualcomm® QCS8450 | 24.781 ms | 13 - 232 MB | NPU |
| Segformer-Base | ONNX | float | Snapdragon® 8 Elite Gen 5 Mobile | 13.269 ms | 21 - 211 MB | NPU |
| Segformer-Base | ONNX | float | Snapdragon® 8 Elite Mobile | 14.413 ms | 21 - 208 MB | NPU |
| Segformer-Base | ONNX | float | Qualcomm® Dragonwing™ IQ-9075 | 25.41 ms | 23 - 26 MB | NPU |
| Segformer-Base | ONNX | float | Qualcomm® Dragonwing™ Q-8750 | 14.413 ms | 21 - 208 MB | NPU |
| Segformer-Base | ONNX | float | Qualcomm® Dragonwing™ IQ-X7181 | 24.584 ms | 35 - 35 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® X2 Elite | 5.62 ms | 18 - 18 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® X Elite | 13.445 ms | 17 - 17 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® 8 Gen 3 Mobile | 9.396 ms | 14 - 247 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 13.313 ms | 9 - 16 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® 8 Elite Mobile | 7.251 ms | 14 - 217 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ IQ-9075 | 14.461 ms | 14 - 16 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® 7 Gen 4 Mobile | 17.071 ms | 14 - 236 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Snapdragon® 8 Elite Gen 5 Mobile | 5.669 ms | 13 - 221 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ Q-6690 | 91.179 ms | 13 - 244 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ Q-7790 | 17.071 ms | 14 - 236 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ Q-8750 | 7.251 ms | 14 - 217 MB | NPU |
| Segformer-Base | ONNX | w8a16 | Qualcomm® Dragonwing™ IQ-X7181 | 13.445 ms | 17 - 17 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® X2 Elite | 1.949 ms | 9 - 9 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® X Elite | 4.99 ms | 5 - 5 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® 8 Gen 3 Mobile | 3.227 ms | 0 - 214 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® 8 Gen 1 Mobile | 7.033 ms | 4 - 221 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ QCS6490 | 19.805 ms | 6 - 9 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 4.753 ms | 5 - 12 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® QCS8450 | 7.033 ms | 4 - 221 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® 8 Elite Mobile | 2.387 ms | 0 - 193 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® 7 Gen 4 Mobile | 5.393 ms | 9 - 206 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ IQ-9075 | 5.116 ms | 6 - 8 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Snapdragon® 8 Elite Gen 5 Mobile | 1.91 ms | 0 - 196 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ Q-6690 | 29.703 ms | 7 - 236 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ Q-7790 | 5.393 ms | 9 - 206 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ Q-8750 | 2.387 ms | 0 - 193 MB | NPU |
| Segformer-Base | ONNX | w8a8 | Qualcomm® Dragonwing™ IQ-X7181 | 4.99 ms | 5 - 5 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® X2 Elite | 12.602 ms | 3 - 3 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® X Elite | 23.529 ms | 3 - 3 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® 8 Gen 3 Mobile | 16.421 ms | 0 - 224 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® 8 Gen 1 Mobile | 29.357 ms | 3 - 226 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® QCS8275 | 48.882 ms | 0 - 190 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 22.717 ms | 3 - 22 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® SA8775P | 24.418 ms | 0 - 190 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® SA8650P | 24.418 ms | 0 - 190 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® SA8255P | 24.418 ms | 0 - 190 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® QCS8450 | 29.357 ms | 3 - 226 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® 8 Elite Gen 5 Mobile | 13.722 ms | 3 - 192 MB | NPU |
| Segformer-Base | QNN_DLC | float | Snapdragon® 8 Elite Mobile | 13.125 ms | 0 - 197 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® Dragonwing™ IQ-9075 | 25.004 ms | 3 - 17 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® SA7255P | 48.882 ms | 0 - 190 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® SA8295P | 27.859 ms | 0 - 191 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® Dragonwing™ Q-8750 | 13.125 ms | 0 - 197 MB | NPU |
| Segformer-Base | QNN_DLC | float | Qualcomm® Dragonwing™ IQ-X7181 | 23.529 ms | 3 - 3 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® X2 Elite | 8.956 ms | 2 - 2 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® X Elite | 19.504 ms | 2 - 2 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® 8 Gen 3 Mobile | 15.2 ms | 0 - 239 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® QCS8275 | 32.41 ms | 2 - 204 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 20.429 ms | 2 - 3 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® SA8775P | 20.2 ms | 2 - 206 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® SA8650P | 20.2 ms | 2 - 206 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® SA8255P | 20.2 ms | 2 - 206 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® 8 Elite Mobile | 11.232 ms | 2 - 208 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ IQ-9075 | 20.443 ms | 2 - 9 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® 7 Gen 4 Mobile | 34.593 ms | 2 - 232 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Snapdragon® 8 Elite Gen 5 Mobile | 9.75 ms | 2 - 217 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ Q-6690 | 115.575 ms | 2 - 254 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ Q-7790 | 34.593 ms | 2 - 232 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ Q-8750 | 11.232 ms | 2 - 208 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® Dragonwing™ IQ-X7181 | 19.504 ms | 2 - 2 MB | NPU |
| Segformer-Base | QNN_DLC | w8a16 | Qualcomm® SA7255P | 32.41 ms | 2 - 204 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® X2 Elite | 2.526 ms | 1 - 1 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® X Elite | 6.326 ms | 1 - 1 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® 8 Gen 3 Mobile | 3.941 ms | 0 - 196 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® 8 Gen 1 Mobile | 8.58 ms | 0 - 199 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ QCS6490 | 21.759 ms | 3 - 7 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® QCS8275 | 10.872 ms | 1 - 174 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 5.766 ms | 0 - 18 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® SA8775P | 6.419 ms | 1 - 175 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® SA8650P | 6.419 ms | 1 - 175 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® SA8255P | 6.419 ms | 1 - 175 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® QCS8450 | 8.58 ms | 0 - 199 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® 8 Elite Mobile | 2.854 ms | 0 - 173 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® 7 Gen 4 Mobile | 6.832 ms | 0 - 178 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ IQ-9075 | 6.131 ms | 1 - 5 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® SA7255P | 10.872 ms | 1 - 174 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® SA8295P | 7.788 ms | 0 - 172 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Snapdragon® 8 Elite Gen 5 Mobile | 2.239 ms | 1 - 183 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ Q-6690 | 28.937 ms | 1 - 188 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ Q-7790 | 6.832 ms | 0 - 178 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ Q-8750 | 2.854 ms | 0 - 173 MB | NPU |
| Segformer-Base | QNN_DLC | w8a8 | Qualcomm® Dragonwing™ IQ-X7181 | 6.326 ms | 1 - 1 MB | NPU |
| Segformer-Base | TFLITE | float | Snapdragon® 8 Gen 3 Mobile | 16.51 ms | 8 - 236 MB | NPU |
| Segformer-Base | TFLITE | float | Snapdragon® 8 Gen 1 Mobile | 29.265 ms | 9 - 233 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® QCS8275 | 48.989 ms | 9 - 200 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 22.681 ms | 9 - 104 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® SA8775P | 24.499 ms | 0 - 192 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® SA8650P | 24.499 ms | 0 - 192 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® SA8255P | 24.499 ms | 0 - 192 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® QCS8450 | 29.265 ms | 9 - 233 MB | NPU |
| Segformer-Base | TFLITE | float | Snapdragon® 8 Elite Gen 5 Mobile | 13.679 ms | 10 - 201 MB | NPU |
| Segformer-Base | TFLITE | float | Snapdragon® 8 Elite Mobile | 13.18 ms | 9 - 203 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® Dragonwing™ IQ-9075 | 25.671 ms | 8 - 30 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® SA7255P | 48.989 ms | 9 - 200 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® SA8295P | 27.81 ms | 9 - 202 MB | NPU |
| Segformer-Base | TFLITE | float | Qualcomm® Dragonwing™ Q-8750 | 13.18 ms | 9 - 203 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Snapdragon® 8 Gen 3 Mobile | 7.104 ms | 2 - 213 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Snapdragon® 8 Gen 1 Mobile | 12.1 ms | 2 - 210 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ QCS6490 | 134.565 ms | 15 - 50 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® QCS8275 | 18.268 ms | 2 - 178 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ QCS8550 (Proxy) | 10.127 ms | 2 - 5 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® SA8775P | 10.881 ms | 2 - 183 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® SA8650P | 10.881 ms | 2 - 183 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® SA8255P | 10.881 ms | 2 - 183 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® QCS8450 | 12.1 ms | 2 - 210 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Snapdragon® 8 Elite Mobile | 5.712 ms | 3 - 179 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Snapdragon® 7 Gen 4 Mobile | 38.939 ms | 15 - 73 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ IQ-9075 | 10.617 ms | 2 - 12 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® SA7255P | 18.268 ms | 2 - 178 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® SA8295P | 12.85 ms | 2 - 187 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Snapdragon® 8 Elite Gen 5 Mobile | 4.553 ms | 1 - 179 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ Q-6690 | 96.343 ms | 13 - 187 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ Q-7790 | 38.939 ms | 15 - 73 MB | NPU |
| Segformer-Base | TFLITE | w8a8 | Qualcomm® Dragonwing™ Q-8750 | 5.712 ms | 3 - 179 MB | NPU |
License
- The license for the original implementation of Segformer-Base can be found here.
References
- SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers
- Source Model Implementation
Community
- Join our AI Hub Slack community to collaborate, post questions and learn more about on-device AI.
- For questions or feedback please reach out to us.
