Fixed image size constant

Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 16:52:47 +01:00
parent b7825c0095
commit 5a7bd6ea9d
1 changed files with 1 additions and 1 deletions
--- 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