From aaec64786c0ca0bca3b0f74dc0b52795a70e36db Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Jul 2019 15:55:01 +0200 Subject: [PATCH] Added debug print of class_id 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 ae8763f..cf7d446 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -91,6 +91,7 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, xmax *= image_size ymax *= image_size + print(class_id) color = colors[class_id] label = f"{classes_to_names[class_id]}" current_axis.add_patch(