dlr: debug

This commit is contained in:
Starbeamrainbowlabs 2023-01-12 18:35:29 +00:00
parent 20f7d34fd1
commit 1b5bb14d8f
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -262,7 +262,7 @@ def plot_predictions(filepath, input_items, colormap, model):
# label_colourmap = decode_segmentation_masks(input_pair[1], colormap, 2)
prediction_colormap = decode_segmentation_masks(prediction_mask, colormap, 2)
print("DEBUG:plot_predictions INFER", prediction_mask)
print("DEBUG:plot_predictions INFER", str(prediction_mask.numpy().tolist()).replace("], [", "],\n["))
plot_samples_matplotlib(
filepath.replace("$$", str(i)),