Skip to content

AV-Lab/Trajectory_Prediction_Models_Zoo

Repository files navigation

Exploration (SOTA datasets)

python explore.py --dataset nuscenes --data_path /media/nadya/Dataset/datasets/nuScenes --split train --obs_len 4 --pred_len 10 --viz_mode lidar_hd_map --max_scenarios 10

python explore.py --dataset waymo --data_path /media/nadya/Dataset/datasets/WAYMO --split train --obs_len 10 --pred_len 20 --viz_mode hd_map --max_scenarios 10

python explore.py --dataset argoverse1 --data_path /media/nadya/Dataset/datasets/Argoverse_1 --split train --obs_len 10 --pred_len 20 --viz_mode hd_map --max_scenarios 10

python explore.py --dataset argoverse2 --data_path /media/nadya/Dataset/datasets/Argoverse_2 --split valid --obs_len 10 --pred_len 20 --viz_mode trajectory

Exploration (Custom Datasets)

python convert/OPV2V/convert_opv2v.py
/path/to/OPV2V
--splits train valid test

python explore.py --dataset trajzoo_dataset --data_path convert/OPV2V/converted --split test --obs_len 20 --pred_len 30 --viz_mode lidar --max_scenarios 20


python convert/V2V4Real/convert_v2v4real.py
/path/to/V2V4Real
--splits train valid test

python explore.py --dataset trajzoo_dataset --data_path convert/V2V4Real/converted --split train --obs_len 20 --pred_len 30 --viz_mode lidar


python convert/DeepAccident/convert_deepaccident.py
/path/to/DeepAccident
--splits train valid

python explore.py --dataset trajzoo_dataset --data_path convert/DeepAccident/converted --split train --obs_len 20 --pred_len 30 --viz_mode lidar --max_scenarios 20


python convert/EMT/convert_emt.py
/path/to/EMT
--splits train test

About

Contains implementation of several Prediction Pipelines and Models.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages