From a685bba4fa81ce47bc680a279d0c019acc05026b Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 17:26:16 +0200 Subject: [PATCH] Properly reversed encoding of labels to visualize them Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index 6febccf..fcdc5d6 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -78,10 +78,16 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, ymax = instance[4] else: class_id = np.argmax(instance[:-12], axis=0) - xmin = (instance[-12] + instance[-8]) * image_size - ymin = (instance[-11] + instance[-7]) * image_size - xmax = (instance[-10] + instance[-6]) * image_size - ymax = (instance[-9] + instance[-5]) * image_size + instance[-12:-8] *= instance[-4:] # multiply with variances + instance[-11, -9] *= np.expand_dims(instance[-5] - instance[-7], axis=-1) + instance[-12, -10] *= np.expand_dims(instance[-6] - instance[-8], axis=-1) + instance[-12:-8] += instance[-8:-4] + instance[-12:-8] *= image_size + + xmin = instance[-12] + ymin = instance[-11] + xmax = instance[-10] + ymax = instance[-9] if class_id == 0: continue