search

alfredgu001324 / MapUncertaintyPrediction

[CVPR 2024 Award Candidate] Producing and Leveraging Online Map Uncertainty in Trajectory Prediction
https://arxiv.org/abs/2403.16439
Apache License 2.0
135 stars 10 forks source link

IndexError: too many indices for array: array is 3-dimensional, but 4 were indexed #14

Open Chenqingsu1 opened 1 month ago

Chenqingsu1 commented 1 month ago

python vis_std2.py --version mini --dataroot ../nuscenes --split val --trj_pred HiVT --map MapTR --trj_data ../trj_data/maptrv2/val_1/data --base_results /MapUncertaintyPrediction-main/HiVT_modified/result_maptrv2_base.pkl --unc_results /MapUncertaintyPrediction-main/HiVT_modified/result_maptrv2_unc.pkl --boxes /MapUncertaintyPrediction-main/HiVT_modified/bbox.pkl --save_path /MapUncertaintyPrediction-main/HiVT_modified/vis_files

This is the command I executed, but I got this error.

0bf58a5982ff0c5153092e9e6ecc924

I tried to output the matrix shape of hivt trj, and it showed (6, 20, 4).

Why does this error occur, and how can I fix it? thanks，I'm afraid to modify the code randomly.🥲

alfredgu001324 commented 1 month ago

Hi, can you maybe double check how you obtain hivt trj? It should be in 3 seconds, but yours is in 2 seconds, seems like the history to me.

Also, split should be mini_val when specifying version to be mini

Chenqingsu1 commented 1 month ago

python eval.py \ --root ../trj_data/maptrv2 \ --split val \ --method base\ --batch_size 32 \ --ckpt_path /data15/chenjh2309/Zhangboran/MapUncertaintyPrediction-main/HiVT_modified/maptrv2_base.ckpt

I executed test.py and obtained results.pkl

910f9f37bd5272815f8f2ce0c419454

And used it as the base_results parameter in the visualization command.

85df0f1c8e19fa56dc672266f4ff19c
alfredgu001324 commented 1 month ago

To debug, maybe look at what this line outputs?