The visualization bboxes are generated from VAD, thanks for their open-sourcing. You can download them from here. This is just for saving visualization time.
Please run
cd MapTRv2_modified
python vis_std.py \
--version mini \ [mini, trainval]
--dataroot ../nuscenes \
--split mini_val \ [mini_val, val]
--trj_pred HiVT \ [HiVT, DenseTNT]
--map MapTR \ [MapTR, StreamMapNet]
--trj_data ../trj_data/maptr/mini_val/data \
--base_results ../PATH_to_baseline_prediction_results \
--unc_results ../PATH_to_unc_prediction_results \
--boxes bbox.pkl \
--save_path $SAVE_PATH
Example demo