Return SSD back to mode inference_fast and updated ssd_keras

Signed-off-by: Jim Martens <github@2martens.de>
2019-05-01 15:34:00 +02:00
parent c0780ac717
commit a37ae400cc
2 changed files with 2 additions and 2 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -144,7 +144,7 @@ def predict(dataset: tf.data.Dataset,
        })
    else:
        checkpointables.update({
-            'ssd': SSD(mode='training', weights_path=weights_path)
+            'ssd': SSD(mode='inference_fast', weights_path=weights_path)
        })

    if checkpoint_path is not None:
--- a/src/twomartens/masterthesis/ssd_keras
+++ b/src/twomartens/masterthesis/ssd_keras