Stars
[NeurIPS 2024] DeMo: Decoupling Motion Forecasting into Directional Intentions and Dynamic States
[NeurIPS 2023, Spotlight] Rank-N-Contrast: Learning Continuous Representations for Regression
PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
[ICCV21] Official implementation of the "Social NCE: Contrastive Learning of Socially-aware Motion Representations" in PyTorch.
SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Driving
The official repository of the ICCV paper "On Exposing the Challenging Long Tail in Future Prediction of Traffic Actors"
[IEEE-TIV] Official implementation of "Hierarchical Transformer-Based Red-Light Running Prediction Model for Two-Wheelers with Multitask Learning"
MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.
This repository allows for fast computation of two-dimensional Time-To-Collision (2D-TTC).
[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting
[ICCV'2023] Forecast-MAE: Self-supervised Pre-training for Motion Forecasting with Masked Autoencoders
Official GitHub repository for Argoverse dataset
[CoRL-2022] SSL-Lanes: Self-Supervised Learning for Motion Forecasting in Autonomous Driving
[ECCV 2022 Oral] Official Implementation of "Social-SSL: Self-Supervised Cross-Sequence Representation Learning Based on Transformers for Multi-Agent Trajectory Prediction"
Official code for article "LLMLight: Large Language Models as Traffic Signal Control Agents".
[ECCV 2020] Code for "Spatio-Temporal Graph Transformer Networks for Pedestrian Trajectory Prediction"
[ISPRS 2023]Official PyTorch Implementation of "GATraj: A Graph- and Attention-based Multi-Agent Trajectory Prediction Model"
Solution for Waymo Motion Prediction Challenge 2022. Our implementation of MultiPath++
[CVPR2023] Leapfrog Diffusion Model for Stochastic Trajectory Prediction
🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.
PIE: A Large-Scale Dataset and Models for Pedestrian Intention Estimation and Trajectory Prediction
Code accompanying the ECCV 2020 paper "Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data" by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (…
This repo contains the code for paper "Dense reinforcement learning for safety validation of autonomous vehicles"
[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression
Pytorch Implementation for Stepwise Goal-Driven Networks for Trajectory Prediction (RA-L/ICRA2022)