def main(cfg):
root_dir = cfg.root_dir
if cfg.save_dir is not None:
output_dir = cfg.save_dir
else:
output_dir = root_dir
gt_dir = osp.join(root_dir, 'gt')
pred_dir = osp.join(root_dir, 'pred')
if cfg.methods is None:
method_names = os.listdir(pred_dir)
else:
method_names = cfg.methods.split(' ')
if cfg.datasets is None:
dataset_names = os.listdir(gt_dir)
else:
dataset_names = cfg.datasets.split(' ')
threads = []
for method in method_names:
for dataset in dataset_names:
loader = EvalDataset(osp.join(pred_dir, method, dataset),
osp.join(gt_dir, dataset))
thread = Eval_thread(loader, method, dataset, output_dir, cfg.cuda)
threads.append(thread)
for thread in threads:
print(thread.run())
def main(cfg):
# root_dir = cfg.root_dir
# if cfg.save_dir is not None:
# output_dir = cfg.save_dir
# else:
# output_dir = root_dir
output_dir = cfg.save_dir
if not os.path.exists(output_dir):
os.makedirs(output_dir)
gt_dir = cfg.gt_root_dir
pred_dir = cfg.pred_root_dir
if cfg.methods is None:
method_names = os.listdir(pred_dir)
else:
#method_names = cfg.methods.split(' ')
method_names = cfg.methods
if cfg.datasets is None:
dataset_names = os.listdir(gt_dir)
else:
#dataset_names = cfg.datasets.split(' ')
dataset_names = cfg.datasets
threads = []
for dataset in dataset_names:
for method in method_names:
loader = EvalDataset(osp.join(pred_dir, method, dataset),
osp.join(gt_dir, dataset))
thread = Eval_thread(loader, method, dataset, output_dir, cfg.cuda)
threads.append(thread)
for thread in threads:
print(thread.run())