From 3fd7ba73732a1d3140377cb26383114f33324d56 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 9 Jul 2019 13:14:24 +0200 Subject: [PATCH] Added conversion to numpy array Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 905c8fa..0351fb0 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -203,7 +203,7 @@ def predict_keras(generator: callable, # do observation stuff predictions = np.asarray(_get_observations(detections)) else: - predictions = ssd_model.predict_on_batch(x) + predictions = np.asarray(ssd_model.predict_on_batch(x)) print(predictions[:, :, [-2, -1]]) decoded_predictions_batch = ssd_output_decoder.decode_detections_fast(