This repo contains the official implementation of our paper:
AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting
Ye Yuan, Xinshuo Weng, Yanglan Ou, Kris Kitani
ICCV 2021
[website] [paper]
Update (07/22): We are actively cleaning the code and will release it before the camera-ready ddl of ICCV 2021.