Added debug print of class_id
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user