argoverse / argoverse-api

Official GitHub repository for Argoverse dataset
https://www.argoverse.org
Other
830 stars 238 forks source link

How to include the confidences in the forecasting results? #322

Open Cram3r95 opened 1 year ago

Cram3r95 commented 1 year ago

At this moment I am including my multimodal predictions as following:

pred_traj_fake_global_aux = pred_traj_fake_global.squeeze(0).view(ARGOVERSE_NUM_MODES, PRED_LEN, 2) output_all[seq_id] = pred_traj_fake_global_aux.cpu().numpy()

Nevertheless, I would like to know how to include the confidences in this array since ARGOVERSE evaluates the multimodal prediction according to these confidences.