Fixed name of weights file

Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 21:10:34 +02:00
parent 5a80315bf9
commit 3c2eabde2d
1 changed files with 1 additions and 1 deletions
--- 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: