Change image size for SSD to 300

Signed-off-by: Jim Martens <github@2martens.de>
2019-04-29 10:33:27 +02:00
parent 3c2eabde2d
commit d624f06c2a
1 changed files with 1 additions and 1 deletions
--- 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"