From 30b2cb9b288dafef860b031c742b65a81a00d005 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 11 Aug 2019 17:21:59 +0200 Subject: [PATCH] Fixed path for pre-trained weights file Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 96d698d..43b0653 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -702,7 +702,7 @@ def _ssd_test_prepare_paths(args: argparse.Namespace, output_path = f"{output_path}/{args.network}/test/{args.iteration}/" checkpoint_path = f"{weights_path}/{args.network}/train/{args.train_iteration}" if test_pretrained: - weights_file = f"{weights_path}/{args.network}/VGG_coco_SSD_300x300_iter_400000_subsampled.h5" + weights_file = f"{weights_path}/ssd/VGG_coco_SSD_300x300_iter_400000_subsampled.h5" else: weights_file = f"{checkpoint_path}/ssd300_weights.h5"