From ce87be6019de3c84016d0655c43dee147ec0079c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 17:29:35 +0200 Subject: [PATCH] Added conversion to numpy array Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index fcdc5d6..a8e50f9 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -77,6 +77,7 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, xmax = instance[3] ymax = instance[4] else: + instance = np.asarray(instance) class_id = np.argmax(instance[:-12], axis=0) instance[-12:-8] *= instance[-4:] # multiply with variances instance[-11, -9] *= np.expand_dims(instance[-5] - instance[-7], axis=-1)