StateTransformer
Multi‑agent trajectory prediction model for autonomous driving.
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.
Not supported
This model is currently not supported on any IoT chipset.
To see performance metrics for this model on other chipsets, click the button below.
View for other chipsetsTechnical Details
Model checkpoint:pretrained-mixtral-small
Input resolution:1x224x224x58, 1x224x224x58, 1x4x7
Number of parameters:90.7M
Model size (float):348 MB
Applicable Scenarios
- Self driving cars
License
Model:MIT
Supported IoT Devices
- Dragonwing IQ-9075 EVK
- Dragonwing Q-8750
- QCS8275 (Proxy)
- QCS8550 (Proxy)
Supported IoT Chipsets
- Qualcomm® QCS8275 (Proxy)
- Qualcomm® QCS8550 (Proxy)
- Qualcomm® QCS9075
Looking for more? See models created by industry leaders.
Discover Model Makers