NVlabs / CTG

Other
66 stars 9 forks source link

Key Error:'rel_pos_norm' #2

Closed VanHelen closed 8 months ago

VanHelen commented 8 months ago

Hello, Thank you very much for your valuable work. I have trained the CTG model and run rollout of a trained model (closed-loop simulation). But when I tried to parse results for rollout(run scripts/parse_scene_edit_results.py ), it seems to have some key error.

Traceback (most recent call last): File "/home/ubuntu2004/Python_Project/CTG/scripts/parse_scene_edit_results.py", line 422, in parse(args) File "/home/ubuntu2004/Python_Project/CTG/scripts/parse_scene_edit_results.py", line 317, in parse agg_res = parse_single_result(result_path, eval_out_dir, args.save_hist_data, gt_hist_path=args.gt_hist) File "/home/ubuntu2004/Python_Project/CTG/scripts/parse_scene_edit_results.py", line 159, in parse_single_result hist2=np.array(hjson["stats"][k]), KeyError: 'rel_pos_norm'

All keys in hjson['stats']: dict_keys(['velocity', 'lon_accel', 'lat_accel', 'jerk']) All keys in gt_hjson['stats']: dict_keys(['velocity', 'lon_accel', 'lat_accel', 'jerk', 'rel_pos_norm', 'rel_lon_pos_abs', 'rel_lat_pos', 'rel_yaw_abs', 'rel_yaw_vel_abs', 'rel_vel_norm', 'rel_lon_vel', 'rel_lat_vel', 'rel_accel_norm', 'rel_lon_accel', 'rel_lat_accel', 'rel_jerk'])

I have read the corresponding keys under ['stats'] in hjson and gt_hjson, and it seems that the keys in the generated hjson file are inconsistent with gt_hjson.