From 3c2eabde2de8088c1d48a3b495375809f0d3ae12 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 21:10:34 +0200 Subject: [PATCH] Fixed name of weights file Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index bcbb779..38cda80 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -176,7 +176,7 @@ def _ssd_val(args: argparse.Namespace) -> None: batch_size = 16 image_size = 256 use_dropout = False - weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300.h5" + weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5" # load prepared ground truth with open(f"{args.ground_truth_path}/photo_paths.bin", "rb") as file: