Plusleizhou's repositories
ADAPT
Official implementation of the ICCV 2023 paper "ADAPT: Efficient Multi-Agent Trajectory Prediction with Adaptation"
apollo
An open autonomous driving platform
Diffusion-Models-pytorch
Pytorch implementation of Diffusion Models (https://arxiv.org/pdf/2006.11239.pdf)
DIPP
Differentiable Integrated Prediction and Planning Framework for Urban Autonomous Driving https://mczhi.github.io/DIPP/
forecast-mae
[ICCV'2023] Forecast-MAE: Self-supervised Pre-training for Motion Forecasting with Masked Autoencoders
HDGT
[IEEE T-PAMI 2023] Unified heterogeneous transformer-based graph neural network for motion prediction
ngsim_env
Learning human driver models from NGSIM data with imitation learning.
PGP
Code for "Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversals," CoRL 2021.
SceneTransformer
Unofficial implementation of SceneTransformer [ICLR 2022]
STrajNet
Codes for STrajNet: Multi-Model Hierarchical Transformer for Occupancy Flow Field Prediction in Autonomous Driving
tnt_vectornet_with_tensorrt
Vectornet with tensorrt deploy, less than 1ms/object for inference.