Jiashun Wang, Huazhe Xu, Medhini Narasimhan, Xiaolong Wang
We propose a novel framework for multi-person 3D motion trajectory prediction. Our key observation is that a human's action and behaviors may highly depend on the other persons around. Thus, instead of predicting each human pose trajectory in isolation, we introduce a Multi-Range Transformers model which contains of a local-range encoder for individual motion and a global-range encoder for social interactions. The Transformer decoder then performs prediction for each person by taking a corresponding pose as a query which attends to both local and global-range encoder features. Our model not only outperforms state-of-the-art methods on long-term 3D motion prediction, but also generates diverse social interactions. More interestingly, our model can even predict 15-person motion simultaneously by automatically dividing the persons into different interaction groups. Project page with code is available at https://jiashunwang.github.io/MRT/.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Autonomous Vehicles | Expi - common actions split | Average MPJPE (mm) @ 1000 ms | 238 | MRT |
| Autonomous Vehicles | Expi - common actions split | Average MPJPE (mm) @ 200 ms | 58 | MRT |
| Autonomous Vehicles | Expi - common actions split | Average MPJPE (mm) @ 400 ms | 116 | MRT |
| Autonomous Vehicles | Expi - common actions split | Average MPJPE (mm) @ 600 ms | 163 | MRT |
| Autonomous Vehicles | Expi - unseen actions split | Average MPJPE (mm) @ 400 ms | 146 | MRT |
| Autonomous Vehicles | Expi - unseen actions split | Average MPJPE (mm) @ 600 ms | 205 | MRT |
| Autonomous Vehicles | Expi - unseen actions split | Average MPJPE (mm) @ 800 ms | 291 | MRT |
| Motion Forecasting | Expi - common actions split | Average MPJPE (mm) @ 1000 ms | 238 | MRT |
| Motion Forecasting | Expi - common actions split | Average MPJPE (mm) @ 200 ms | 58 | MRT |
| Motion Forecasting | Expi - common actions split | Average MPJPE (mm) @ 400 ms | 116 | MRT |
| Motion Forecasting | Expi - common actions split | Average MPJPE (mm) @ 600 ms | 163 | MRT |
| Motion Forecasting | Expi - unseen actions split | Average MPJPE (mm) @ 400 ms | 146 | MRT |
| Motion Forecasting | Expi - unseen actions split | Average MPJPE (mm) @ 600 ms | 205 | MRT |
| Motion Forecasting | Expi - unseen actions split | Average MPJPE (mm) @ 800 ms | 291 | MRT |
| Autonomous Driving | Expi - common actions split | Average MPJPE (mm) @ 1000 ms | 238 | MRT |
| Autonomous Driving | Expi - common actions split | Average MPJPE (mm) @ 200 ms | 58 | MRT |
| Autonomous Driving | Expi - common actions split | Average MPJPE (mm) @ 400 ms | 116 | MRT |
| Autonomous Driving | Expi - common actions split | Average MPJPE (mm) @ 600 ms | 163 | MRT |
| Autonomous Driving | Expi - unseen actions split | Average MPJPE (mm) @ 400 ms | 146 | MRT |
| Autonomous Driving | Expi - unseen actions split | Average MPJPE (mm) @ 600 ms | 205 | MRT |
| Autonomous Driving | Expi - unseen actions split | Average MPJPE (mm) @ 800 ms | 291 | MRT |