INNER CODE UNIT · Python

main_per_cat

xinshuoweng/AB3DMOT · main.py:21

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 
	eval_dir_dict = dict()
	for index in range(cfg.num_hypo):
		eval_dir_dict[index] = os.path.join(save_dir, 'data_%d' % index); mkdir_if_missing(eval_dir_dict[index]) 		

	# loop every sequence
	seq_count = 0
	total_time, total_frames = 0.0, 0
	for seq_name in seq_eval:

View source record →