From 5a7bd6ea9d7cc9cdfa05993064bd434fb08e7181 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 20 Mar 2019 16:52:47 +0100 Subject: [PATCH] Fixed image size constant Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index c839dab..5922fd7 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -45,7 +45,7 @@ from twomartens.masterthesis.ssd_keras.models import keras_ssd300_dropout K = tf.keras.backend tfe = tf.contrib.eager -IMAGE_SIZE = (240, 320, 3) # TODO check with SceneNet RGB-D +IMAGE_SIZE = (300, 300, 3) N_CLASSES = 80 DROPOUT_RATE = 0.5 IOU_THRESHOLD = 0.45