From a37ae400cc79930ac66111ab62e97311cc01f42d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 15:34:00 +0200 Subject: [PATCH] Return SSD back to mode inference_fast and updated ssd_keras Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- src/twomartens/masterthesis/ssd_keras | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index b2291f6..15e83c9 100644 --- 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: diff --git a/src/twomartens/masterthesis/ssd_keras b/src/twomartens/masterthesis/ssd_keras index ccf74fe..0cb1f77 160000 --- a/src/twomartens/masterthesis/ssd_keras +++ b/src/twomartens/masterthesis/ssd_keras @@ -1 +1 @@ -Subproject commit ccf74fe224cb28c62f9d6d6e5e8c919ff7ed81c9 +Subproject commit 0cb1f7750e76fa6b21dad9ec38eb8fb1cdca0450