meTroyChan

stable-diffusion

A latent text-to-image diffusion model

apollo

An open autonomous driving platform

Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

SensorsCalibration

OpenCalib: A Multi-sensor Calibration Toolbox for Autonomous Driving

Learning-Deep-Learning

Paper reading notes on Deep Learning and Machine Learning

how-do-vits-work

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

nuplan-devkit

The devkit of the nuPlan dataset.

MTR

MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.

HiVT

[CVPR 2022] HiVT: Hierarchical Vector Transformer for Multi-Agent Motion Prediction

QCNet

[CVPR 2023] Query-Centric Trajectory Prediction

VectorMapNet_code

This is the official code base of VectorMapNet (ICML 2023)

mmTransformer

[CVPR 2021] "Multimodal Motion Prediction with Stacked Transformers": official code implementation and project page.

av2-api

Argoverse 2: Next generation datasets for self-driving perception and forecasting.

trajdata

A unified interface to many trajectory forecasting datasets.

AgentFormer

[ICCV 2021] Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".

VectorNet

VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation, partial function re-implement

VectorNet

Pytorch implementation of CVPR2020 paper “VectorNet: Encoding HD Maps and Agent Dynamics from Vectorized Representation”

tensorrt_plugin_generator

A simple tool that can generate TensorRT plugin code quickly.

PGP

Code for "Multimodal Trajectory Prediction Conditioned on Lane-Graph Traversals," CoRL 2021.

interaction-dataset

Interaction Dataset Python Scripts

M2I

M2I is a simple but effective joint motion prediction framework through marginal and conditional predictions by exploiting the factorized relations between interacting agents.

drone-dataset-tools

The goal of this repo is to make the drone datasets as easy to use as possible. For this purpose we provide source code in Python, which allows the import and visualization of the datasets.

WIMP

[arXiv] What-If Motion Prediction for Autonomous Driving ❓🚗💨

HPTR

Real-Time Motion Prediction via Heterogeneous Polyline Transformer with Relative Pose Encoding. NeurIPS 2023.

kaggle-lyft-motion-prediction-av

The 3rd place solution for competition "Lyft Motion Prediction for Autonomous Vehicles" at Kaggle

TNT-VectorNet-and-HOME-Trajectory-Forecasting

Custom implementation of TNT-VectorNet and HOME architectures

RRB

Official implementation of "Injecting Knowledge in Data-driven Vehicle Trajectory Predictors", Published in Transportation research part C.

constant_velocity_pedestrian_motion

Code for "What the Constant Velocity Model Can Teach Us About Pedestrian Motion Prediction", Schöller et al., Robotics and Automation Letters (RA-L), 2020

adaptive-prediction

VectorNet_Waymo

Waymo motion prediction using VectorNet