Added debug print of class_id

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-10 15:55:01 +02:00
parent 7969716e6d
commit aaec64786c

View File

@ -91,6 +91,7 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray,
xmax *= image_size xmax *= image_size
ymax *= image_size ymax *= image_size
print(class_id)
color = colors[class_id] color = colors[class_id]
label = f"{classes_to_names[class_id]}" label = f"{classes_to_names[class_id]}"
current_axis.add_patch( current_axis.add_patch(