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 IoT Devices

  • QCS8275 (Proxy)
  • QCS8550 (Proxy)
  • QCS9075 (Proxy)

Supported IoT Chipsets

  • Qualcomm® QCS8275 (Proxy)
  • Qualcomm® QCS8550 (Proxy)
  • Qualcomm® QCS9075 (Proxy)

Looking for more? See models created by industry leaders.

Discover Model Makers