search

NVlabs / ScePT

Code for the CVPR 2022 paper "ScePT: Scene-consistent, Policy-based Trajectory Predictions for Planning" by Yuxiao Chen, Boris Ivanovic, and Marco Pavone
Other
81 stars 17 forks source link

Mode Probabilities #12

Open ologandavid opened 8 months ago

ologandavid commented 8 months ago

How would I go about extracting the probabilities associated which each mode in the prediction?

Thanks, David

chenyx09 commented 8 months ago

Hi David, please check https://github.com/NVlabs/ScePT/blob/main/ScePT/model/mgcvae_clique.py#L1433, the pi_list is the probability of modes.