From d624f06c2a4117395fa30f4bf220d0a2ecc86760 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 29 Apr 2019 10:33:27 +0200 Subject: [PATCH] Change image size for SSD to 300 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 38cda80..93e1aeb 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -174,7 +174,7 @@ def _ssd_val(args: argparse.Namespace) -> None: tf.enable_eager_execution() batch_size = 16 - image_size = 256 + image_size = 300 use_dropout = False weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5"