From 1a0398191a7406850a37f887a4aa6cc930dadc1a Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 18 Jul 2019 12:46:57 +0200 Subject: [PATCH] Removed checkpoint path from prepared paths for ssd test Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 37d8e59..2891314 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -228,8 +228,8 @@ def _ssd_test(args: argparse.Namespace) -> None: use_dropout = _ssd_is_dropout(args) - output_path, checkpoint_path, weights_file = _ssd_test_prepare_paths(args, output_path, - weights_path, test_pretrained) + output_path, weights_file = _ssd_test_prepare_paths(args, output_path, + weights_path, test_pretrained) file_names, instances = _ssd_test_load_gt(ground_truth_path) @@ -612,7 +612,7 @@ def _ssd_train_prepare_paths(args: argparse.Namespace, def _ssd_test_prepare_paths(args: argparse.Namespace, output_path: str, weights_path: str, - test_pretrained: bool) -> Tuple[str, str, str]: + test_pretrained: bool) -> Tuple[str, str]: import os output_path = f"{output_path}/{args.network}/test/{args.iteration}/" @@ -624,7 +624,7 @@ def _ssd_test_prepare_paths(args: argparse.Namespace, os.makedirs(output_path, exist_ok=True) - return output_path, checkpoint_path, weights_file + return output_path, weights_file def _ssd_evaluate_prepare_paths(args: argparse.Namespace,