From 505ea330fc44978b2f96f95505a3533f90b61403 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 25 Jul 2019 15:30:54 +0200 Subject: [PATCH] Added missing creation of visualisation output directory Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index e6430a9..daa11df 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -752,10 +752,13 @@ def _measure_prepare_paths(args: argparse.Namespace, def _visualise_prepare_paths(args: argparse.Namespace, output_path: str, coco_path: str, gt_path: str) -> Tuple[str, str, str]: + import os output_path = f"{output_path}/visualise/{args.trajectory}" annotation_file_train = f"{coco_path}/annotations/instances_train2014.json" 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