From e602e7339d53eb9aa7d87a3c75cf9f1b860b00d5 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 10 Jun 2019 11:21:35 +0200 Subject: [PATCH] Provided checkpoint path in validation case Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 404e2cc..4e97cdc 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -242,6 +242,7 @@ def _ssd_val(args: argparse.Namespace) -> None: use_dropout = False if args.network == "ssd" else True weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5" + checkpoint_path = f"{args.weights_path}/train/{args.network}/" output_path = f"{args.output_path}/val/{args.network}/{args.iteration}/" os.makedirs(output_path, exist_ok=True) @@ -262,11 +263,11 @@ def _ssd_val(args: argparse.Namespace) -> None: ) if args.debug: with use_summary_writer.as_default(): - ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits, - forward_passes_per_image=forward_passes_per_image) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, checkpoint_path, + nr_digits=nr_digits, forward_passes_per_image=forward_passes_per_image) else: - ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits, - forward_passes_per_image=forward_passes_per_image) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, checkpoint_path, + nr_digits=nr_digits, forward_passes_per_image=forward_passes_per_image) def _auto_encoder_val(args: argparse.Namespace) -> None: