| import mmcv | |
| from typing import Dict, List | |
| # predroot = '/mnt/hdd/models/UniAD/stage2_e2e/017_carla_closedloop/e2e_carla_slurm_r50_v2t1_paradrive/test/Sun_Apr_14_19_11_00_2024/results.pkl' | |
| predroot = '/mnt/hdd2/datasets/navsim/openscene-v1.1/infos/nuplan_navsim_train.pkl' | |
| outputs: Dict = mmcv.load(predroot) | |
| # keys are bbox_results, occ_results_computed, planning_results_computed | |
| # print(outputs.keys()) | |
| # print(outputs[0].keys()) | |
| # zxc | |
| # outputs: List[Dict] = outputs["infos"] | |
| # outputs: List[Dict] = outputs["bbox_results"] | |
| # self.outputs = outputs | |
| for index in range(len(outputs)): | |
| data = outputs[index] | |
| print('\n\nnew frame') | |
| print('scene_token ', data['scene_token']) | |
| print(data['token']) | |
| print(data['frame_idx']) |