CrazyWangBa

TrackNPred

This is the code base for our ACM CSCS 2019 paper: "RobustTP: End-to-End Trajectory Prediction for Heterogeneous Road-Agents in Dense Traffic with Noisy Sensor Inputs". This codebase contains implementations for several trajectory prediction methods including Social-GAN and TraPHic.

yolov5_train_on_UA-DETRAC

Train YOLOv5 with dataset UA-DETRAC to detect vehicles.

YoloV5_MCMOT

多类别多目标跟踪YoloV5+sort/deepsort/bytetrack/BotSort/motdt

CoordAttention

Code for our CVPR2021 paper coordinate attention

PiP-Planning-informed-Prediction

(ECCV 2020) PiP: Planning-informed Trajectory Prediction for Autonomous Driving

SSL-Lanes

[CoRL-2022] SSL-Lanes: Self-Supervised Learning for Motion Forecasting in Autonomous Driving

crat-pred

FutuTP

FutuTP: Future-based Trajectory Prediction for Autonomous Driving

DSP

Trajectory Prediction with Graph-based Dual-scale Context Fusion

ADAPT

Official implementation of the ICCV 2023 paper "ADAPT: Efficient Multi-Agent Trajectory Prediction with Adaptation"

HLSTrajForecast

The official implementation of "Hierarchical Latent Structure for Multi-Modal Vehicle Trajectory Forecasting" presented in ECCV2022.

HLTP

[IEEE TIV] Official PyTorch Implementation of ''A Cognitive-Based Trajectory Prediction Approach for Autonomous Driving.''

yolov5-ghostnetv2-SIOU-CA-WBF-V6.2

GAT-pt

PyTorch implementation of the Graph Attention Networks (GAT) based on the paper "Graph Attention Network" by Velickovic et al - https://arxiv.org/abs/1710.10903v3

NGSIM-US-101-trajectory-dataset-smoothing

smoothing the NGSIM US-101 trajectory dataset using Savitzky-Golay Filter

BATraj-Behavior-aware-Model

[AAAI 2024] Official PyTorch Implementation of ''BAT: Behavior-Aware Human-Like Trajectory Prediction for Autonomous Driving''.

DGFNet

Multi-Agent Trajectory Prediction with Difficulty-Guided Feature Enhancement Network

CSE251B-Argoverse-Trajectory-Prediction

Argoverse2_Motion_Forecasting

MFTF: Motion Forecasting Using Transformers

deep_prediction

Motion Forecasting for Autonomous Vehicles using the Argoverse Motion Forecasting Dataset

Argoverse-Data-Preprocess

Data preprocessing tool for Argoverse Motion Forecasting Benchmark

LAformer

[CVPRW 2024]Official PyTorch Implementation of "LAformer: Trajectory Prediction for Autonomous Driving with Lane-Aware Scene Constraints"

Improving-Efficiency-and-Generalisability-of-MP

(IEEE-ITS 2023) Improving Efficiency and Generalisability of Motion Prediction with Deep Multi-Agent Learning and Multi-Head Attention

VTPLSTM

Inspired by the performence of Social LSTM. I created a model that could predict vehicle's trajectory in 5s. The model uses LSTM as center. I trained it at NGSIM. Still fix it...

ChatGLM2_Lora

This repo was a simple way to implement Lora to fine-tuning ChatGLM2.这个项目是用LORA微调chatglm2的简单实现。