File size: 904 Bytes
f693366 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import os
import tqdm
import json
from visual_nuscenes import NuScenes
use_gt = False
out_dir = './result_vis/'
result_json = "work_dirs/pp-nus/results_eval/pts_bbox/results_nusc"
dataroot='/data/nuscenes'
if not os.path.exists(out_dir):
os.mkdir(out_dir)
if use_gt:
nusc = NuScenes(version='v1.0-trainval', dataroot=dataroot, verbose=True, pred = False, annotations = "sample_annotation")
else:
nusc = NuScenes(version='v1.0-trainval', dataroot=dataroot, verbose=True, pred = True, annotations = result_json, score_thr=0.25)
with open('{}.json'.format(result_json)) as f:
table = json.load(f)
tokens = list(table['results'].keys())
for token in tqdm.tqdm(tokens[:100]):
if use_gt:
nusc.render_sample(token, out_path = "./result_vis/"+token+"_gt.png", verbose=False)
else:
nusc.render_sample(token, out_path = "./result_vis/"+token+"_pred.png", verbose=False)
|