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.
Technical 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
Licenses
Deployable Model:AI-HUB-MODELS-LICENSE
Supported Compute Devices
- Snapdragon X Elite CRD
- Snapdragon X Plus 8-Core CRD
Supported Compute Chipsets
- Snapdragon® X Elite
- Snapdragon® X Plus 8-Core
Looking for more? See models created by industry leaders.
Discover Model Makers








