From c71fb889dafc7abfa86a9bdaffdc6bfa5a8c39fa Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 29 Apr 2019 11:04:17 +0200 Subject: [PATCH] Added conversion to numpy arrays 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 3ccaead..6a16926 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -180,9 +180,9 @@ def _predict_one_epoch(dataset: tf.data.Dataset, if use_dropout: for _ in range(forward_passes_per_image): decoded_predictions_pass = ssd(inputs) - decoded_predictions_batch.append(decoded_predictions_pass) + decoded_predictions_batch.append(np.array(decoded_predictions_pass)) else: - decoded_predictions_batch.append(ssd(inputs)) + decoded_predictions_batch.append(np.array(ssd(inputs))) # save predictions batch-wise to prevent memory problems print(decoded_predictions_batch[0])