From c2d8da39c88dc145c52a506ba32d31a692fe8a6c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 16:11:14 +0200 Subject: [PATCH] Fixed calculation of bbox coordinates Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index 08faaca..6febccf 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -78,19 +78,14 @@ 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] - ymin = instance[-11] + instance[-7] - xmax = instance[-10] + instance[-6] - ymax = instance[-9] + instance[-5] + 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 if class_id == 0: continue - xmin *= image_size - ymin *= image_size - xmax *= image_size - ymax *= image_size - color = colors[class_id] label = f"{classes_to_names[class_id]}" current_axis.add_patch(