qaihm-bot commited on
Commit
e782210
·
verified ·
1 Parent(s): 90b8572

See https://github.com/qualcomm/ai-hub-models/releases/v0.49.1 for changelog.

Files changed (1) hide show
  1. README.md +15 -14
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/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,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.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,15 +57,16 @@ See our repository for [StateTransformer on GitHub](https://github.com/qualcomm/
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
 
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/tree/v0.49.1/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.49.1/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.49.1/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/tree/v0.49.1/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/tree/v0.49.1/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 | 837.502 ms | 204 - 204 MB | NPU
61
+ | StateTransformer | ONNX | float | Snapdragon® X Elite | 1329.236 ms | 184 - 184 MB | NPU
62
+ | StateTransformer | ONNX | float | Snapdragon® 8 Gen 3 Mobile | 674.61 ms | 100 - 5306 MB | NPU
63
+ | StateTransformer | ONNX | float | Snapdragon® 8 Elite For Galaxy Mobile | 606.06 ms | 89 - 2117 MB | NPU
64
+ | StateTransformer | TFLITE | float | Snapdragon® 8 Elite Gen 5 Mobile | 357.44 ms | 225 - 247 MB | CPU
65
+ | StateTransformer | TFLITE | float | Snapdragon® 8 Gen 3 Mobile | 612.043 ms | 220 - 249 MB | CPU
66
+ | StateTransformer | TFLITE | float | Qualcomm® QCS8275 (Proxy) | 995.373 ms | 223 - 237 MB | CPU
67
+ | StateTransformer | TFLITE | float | Qualcomm® QCS8550 (Proxy) | 601.94 ms | 172 - 293 MB | CPU
68
+ | StateTransformer | TFLITE | float | Qualcomm® QCS8450 (Proxy) | 711.162 ms | 217 - 235 MB | CPU
69
+ | StateTransformer | TFLITE | float | Snapdragon® 8 Elite For Galaxy Mobile | 408.178 ms | 159 - 169 MB | CPU
70
 
71
  ## License
72
  * The license for the original implementation of StateTransformer can be found