v0.48.0
Browse filesSee https://github.com/qualcomm/ai-hub-models/releases/v0.48.0 for changelog.
README.md
CHANGED
|
@@ -14,7 +14,7 @@ pipeline_tag: other
|
|
| 14 |
|
| 15 |
StateTransformer is a transformer-based model designed for trajectory prediction in self-driving scenarios. It integrates rasterized map data, agent context, and temporal dynamics to generate accurate future trajectories.
|
| 16 |
|
| 17 |
-
This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the [Qualcomm® AI Hub Models](https://github.com/
|
| 18 |
|
| 19 |
Qualcomm AI Hub Models uses [Qualcomm AI Hub Workbench](https://workbench.aihub.qualcomm.com) to compile, profile, and evaluate this model. [Sign up](https://myaccount.qualcomm.com/signup) to run these models on a hosted Qualcomm® device.
|
| 20 |
|
|
@@ -27,22 +27,22 @@ Below are pre-exported model assets ready for deployment.
|
|
| 27 |
|
| 28 |
| Runtime | Precision | Chipset | SDK Versions | Download |
|
| 29 |
|---|---|---|---|---|
|
| 30 |
-
| ONNX | float | Universal | QAIRT 2.42, ONNX Runtime 1.24.1 | [Download](https://qaihub-public-assets.s3.us-west-2.amazonaws.com/qai-hub-models/models/statetransformer/releases/v0.
|
| 31 |
-
| TFLITE | float | Universal | TFLite 2.17.0 | [Download](https://qaihub-public-assets.s3.us-west-2.amazonaws.com/qai-hub-models/models/statetransformer/releases/v0.
|
| 32 |
|
| 33 |
For more device-specific assets and performance metrics, visit **[StateTransformer on Qualcomm® AI Hub](https://aihub.qualcomm.com/models/statetransformer)**.
|
| 34 |
|
| 35 |
|
| 36 |
### Option 2: Export with Custom Configurations
|
| 37 |
|
| 38 |
-
Use the [Qualcomm® AI Hub Models](https://github.com/
|
| 39 |
- Custom weights (e.g., fine-tuned checkpoints)
|
| 40 |
- Custom input shapes
|
| 41 |
- Target device and runtime configurations
|
| 42 |
|
| 43 |
This option is ideal if you need to customize the model beyond the default configuration provided here.
|
| 44 |
|
| 45 |
-
See our repository for [StateTransformer on GitHub](https://github.com/
|
| 46 |
|
| 47 |
## Model Details
|
| 48 |
|
|
@@ -57,16 +57,15 @@ See our repository for [StateTransformer on GitHub](https://github.com/quic/ai-h
|
|
| 57 |
## Performance Summary
|
| 58 |
| Model | Runtime | Precision | Chipset | Inference Time (ms) | Peak Memory Range (MB) | Primary Compute Unit
|
| 59 |
|---|---|---|---|---|---|---
|
| 60 |
-
| StateTransformer | ONNX | float | Snapdragon®
|
| 61 |
-
| StateTransformer | ONNX | float | Snapdragon®
|
| 62 |
-
| StateTransformer | ONNX | float | Snapdragon® 8 Elite For Galaxy Mobile |
|
| 63 |
-
| StateTransformer |
|
| 64 |
-
| StateTransformer | TFLITE | float |
|
| 65 |
-
| StateTransformer | TFLITE | float | Qualcomm®
|
| 66 |
-
| StateTransformer | TFLITE | float | Qualcomm®
|
| 67 |
-
| StateTransformer | TFLITE | float |
|
| 68 |
-
| StateTransformer | TFLITE | float | Snapdragon® 8 Elite
|
| 69 |
-
| StateTransformer | TFLITE | float | Snapdragon® 8 Elite Gen 5 Mobile | 348.775 ms | 226 - 246 MB | CPU
|
| 70 |
|
| 71 |
## License
|
| 72 |
* The license for the original implementation of StateTransformer can be found
|
|
|
|
| 14 |
|
| 15 |
StateTransformer is a transformer-based model designed for trajectory prediction in self-driving scenarios. It integrates rasterized map data, agent context, and temporal dynamics to generate accurate future trajectories.
|
| 16 |
|
| 17 |
+
This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the [Qualcomm® AI Hub Models](https://github.com/qualcomm/ai-hub-models/blob/main/qai_hub_models/models/statetransformer) library to export with custom configurations. More details on model performance across various devices, can be found [here](#performance-summary).
|
| 18 |
|
| 19 |
Qualcomm AI Hub Models uses [Qualcomm AI Hub Workbench](https://workbench.aihub.qualcomm.com) to compile, profile, and evaluate this model. [Sign up](https://myaccount.qualcomm.com/signup) to run these models on a hosted Qualcomm® device.
|
| 20 |
|
|
|
|
| 27 |
|
| 28 |
| Runtime | Precision | Chipset | SDK Versions | Download |
|
| 29 |
|---|---|---|---|---|
|
| 30 |
+
| ONNX | float | Universal | QAIRT 2.42, ONNX Runtime 1.24.1 | [Download](https://qaihub-public-assets.s3.us-west-2.amazonaws.com/qai-hub-models/models/statetransformer/releases/v0.48.0/statetransformer-onnx-float.zip)
|
| 31 |
+
| TFLITE | float | Universal | TFLite 2.17.0 | [Download](https://qaihub-public-assets.s3.us-west-2.amazonaws.com/qai-hub-models/models/statetransformer/releases/v0.48.0/statetransformer-tflite-float.zip)
|
| 32 |
|
| 33 |
For more device-specific assets and performance metrics, visit **[StateTransformer on Qualcomm® AI Hub](https://aihub.qualcomm.com/models/statetransformer)**.
|
| 34 |
|
| 35 |
|
| 36 |
### Option 2: Export with Custom Configurations
|
| 37 |
|
| 38 |
+
Use the [Qualcomm® AI Hub Models](https://github.com/qualcomm/ai-hub-models/blob/main/qai_hub_models/models/statetransformer) Python library to compile and export the model with your own:
|
| 39 |
- Custom weights (e.g., fine-tuned checkpoints)
|
| 40 |
- Custom input shapes
|
| 41 |
- Target device and runtime configurations
|
| 42 |
|
| 43 |
This option is ideal if you need to customize the model beyond the default configuration provided here.
|
| 44 |
|
| 45 |
+
See our repository for [StateTransformer on GitHub](https://github.com/qualcomm/ai-hub-models/blob/main/qai_hub_models/models/statetransformer) for usage instructions.
|
| 46 |
|
| 47 |
## Model Details
|
| 48 |
|
|
|
|
| 57 |
## Performance Summary
|
| 58 |
| Model | Runtime | Precision | Chipset | Inference Time (ms) | Peak Memory Range (MB) | Primary Compute Unit
|
| 59 |
|---|---|---|---|---|---|---
|
| 60 |
+
| StateTransformer | ONNX | float | Snapdragon® X2 Elite | 825.712 ms | 205 - 205 MB | NPU
|
| 61 |
+
| StateTransformer | ONNX | float | Snapdragon® X Elite | 1335.801 ms | 184 - 184 MB | NPU
|
| 62 |
+
| StateTransformer | ONNX | float | Snapdragon® 8 Elite For Galaxy Mobile | 593.726 ms | 93 - 2120 MB | NPU
|
| 63 |
+
| StateTransformer | TFLITE | float | Snapdragon® 8 Gen 3 Mobile | 519.387 ms | 222 - 240 MB | CPU
|
| 64 |
+
| StateTransformer | TFLITE | float | Qualcomm® QCS8275 (Proxy) | 1003.679 ms | 226 - 242 MB | CPU
|
| 65 |
+
| StateTransformer | TFLITE | float | Qualcomm® QCS8550 (Proxy) | 566.153 ms | 163 - 236 MB | CPU
|
| 66 |
+
| StateTransformer | TFLITE | float | Qualcomm® QCS8450 (Proxy) | 773.28 ms | 216 - 238 MB | CPU
|
| 67 |
+
| StateTransformer | TFLITE | float | Snapdragon® 8 Elite For Galaxy Mobile | 400.564 ms | 202 - 224 MB | CPU
|
| 68 |
+
| StateTransformer | TFLITE | float | Snapdragon® 8 Elite Gen 5 Mobile | 351.862 ms | 226 - 247 MB | CPU
|
|
|
|
| 69 |
|
| 70 |
## License
|
| 71 |
* The license for the original implementation of StateTransformer can be found
|