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 Mobile Devices
- Samsung Galaxy S21
- Samsung Galaxy S21 Ultra
- Samsung Galaxy S21+
- Samsung Galaxy S22 5G
- Samsung Galaxy S22 Ultra 5G
- Samsung Galaxy S22+ 5G
- Samsung Galaxy S23
- Samsung Galaxy S23 Ultra
- Samsung Galaxy S23+
- Samsung Galaxy S24
- Samsung Galaxy S24 Ultra
- Samsung Galaxy S24+
- Samsung Galaxy S25
- Samsung Galaxy S25 Ultra
- Samsung Galaxy S25+
- Samsung Galaxy Tab S8
- Snapdragon 8 Elite Gen 5 QRD
- Xiaomi 12
- Xiaomi 12 Pro
Supported Mobile Chipsets
- Snapdragon® 8 Elite For Galaxy Mobile
- Snapdragon® 8 Elite Gen5 Mobile
- Snapdragon® 8 Gen 1 Mobile
- Snapdragon® 8 Gen 2 Mobile
- Snapdragon® 8 Gen 3 Mobile
- Snapdragon® 888 Mobile
Looking for more? See models created by industry leaders.
Discover Model Makers