Added missing creation of visualisation output directory

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-25 15:30:54 +02:00
parent 45b5fda185
commit 505ea330fc

View File

@ -752,11 +752,14 @@ def _measure_prepare_paths(args: argparse.Namespace,
def _visualise_prepare_paths(args: argparse.Namespace, def _visualise_prepare_paths(args: argparse.Namespace,
output_path: str, coco_path: str, output_path: str, coco_path: str,
gt_path: str) -> Tuple[str, str, str]: gt_path: str) -> Tuple[str, str, str]:
import os
output_path = f"{output_path}/visualise/{args.trajectory}" output_path = f"{output_path}/visualise/{args.trajectory}"
annotation_file_train = f"{coco_path}/annotations/instances_train2014.json" annotation_file_train = f"{coco_path}/annotations/instances_train2014.json"
ground_truth_path = f"{gt_path}/{args.tarball_id}/" ground_truth_path = f"{gt_path}/{args.tarball_id}/"
os.makedirs(output_path, exist_ok=True)
return output_path, annotation_file_train, ground_truth_path return output_path, annotation_file_train, ground_truth_path