2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Removed checkpoint path from prepared paths for ssd test

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-18 12:46:57 +02:00
parent 015befd542
commit 1a0398191a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/twomartens/masterthesis/cli.py
View File

@ -228,7 +228,7 @@ def _ssd_test(args: argparse.Namespace) -> None:
use_dropout = _ssd_is_dropout(args) use_dropout = _ssd_is_dropout(args)
output_path, checkpoint_path, weights_file = _ssd_test_prepare_paths(args, output_path, output_path, weights_file = _ssd_test_prepare_paths(args, output_path,
weights_path, test_pretrained) weights_path, test_pretrained)
file_names, instances = _ssd_test_load_gt(ground_truth_path) 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, def _ssd_test_prepare_paths(args: argparse.Namespace,
output_path: str, weights_path: str, output_path: str, weights_path: str,
test_pretrained: bool) -> Tuple[str, str, str]: test_pretrained: bool) -> Tuple[str, str]:
import os import os
output_path = f"{output_path}/{args.network}/test/{args.iteration}/" 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) 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, def _ssd_evaluate_prepare_paths(args: argparse.Namespace,