File size: 745 Bytes
663494c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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']) |