From db53c35929b6fa617861418a75725a5a23a0f54b Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 9 Jul 2019 15:34:05 +0200 Subject: [PATCH] Skip background boxes when saving training images Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index b10395b..8a565d6 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -60,6 +60,8 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, output_path: s xmax = (instance[-10] + instance[-6]) * image_size ymax = (instance[-9] + instance[-5]) * image_size class_id = np.argmax(instance[:-12], axis=0) + if class_id == 0: + continue color = colors[class_id] label = f"{classes_to_names[class_id]}" current_axis.add_patch(