Added debug print of class_id

Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 15:55:01 +02:00
parent 7969716e6d
commit aaec64786c
1 changed files with 1 additions and 0 deletions
--- 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(