From 72c821dc2aa3ceecfcccaaa26344f2522eeafcb7 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 11 Aug 2019 13:49:46 +0200 Subject: [PATCH] Changed pre-trained weights file to sub-sampled one 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 e2330c6..f64c178 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.h5" + weights_file = f"{weights_path}/{args.network}/VGG_coco_SSD_300x300_iter_400000_subsampled.h5" else: weights_file = f"{checkpoint_path}/ssd300_weights.h5"