2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed image size constant

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-03-20 16:52:47 +01:00
parent b7825c0095
commit 5a7bd6ea9d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/ssd.py
View File

@ -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