From 01a09ba474415f6ffc34bf995ee0fec4aae32b2e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 9 Jul 2019 12:23:27 +0200 Subject: [PATCH] Fixed SSD prediction Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 7eca237..92bd2e0 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -190,7 +190,7 @@ def predict_keras(generator: callable, detections = None batch_size = None for _ in range(forward_passes_per_image): - predictions = ssd_model.model.predict_on_batch(x) + predictions = ssd_model.predict_on_batch(x) if batch_size is None: batch_size = predictions.shape[0] if detections is None: @@ -203,7 +203,7 @@ def predict_keras(generator: callable, # do observation stuff predictions = np.asarray(_get_observations(detections)) else: - predictions = ssd_model.model.predict_on_batch(x) + predictions = ssd_model.predict_on_batch(x) decoded_predictions_batch = ssd_output_decoder.decode_detections_fast( y_pred=predictions,