INNER CODE UNIT · Python
parse_args
xinshuoweng/AB3DMOT · main.py:13
def parse_args():
parser = argparse.ArgumentParser(description='AB3DMOT')
parser.add_argument('--dataset', type=str, default='nuScenes', help='KITTI, nuScenes')
parser.add_argument('--split', type=str, default='', help='train, val, test')
parser.add_argument('--det_name', type=str, default='', help='pointrcnn')
args = parser.parse_args()
return args
def main_per_cat(cfg, cat, log, ID_start):
# get data-cat-split specific path
result_sha = '%s_%s_%s' % (cfg.det_name, cat, cfg.split)
det_root = os.path.join('./data', cfg.dataset, 'detection', result_sha)
subfolder, det_id2str, hw, seq_eval, data_root = get_subfolder_seq(cfg.dataset, cfg.split)
trk_root = os.path.join(data_root, 'tracking')
save_dir = os.path.join(cfg.save_root, result_sha + '_H%d' % cfg.num_hypo); mkdir_if_missing(save_dir)
# create eval dir for each hypothesis