dlr predict: comment debug

This commit is contained in:
Starbeamrainbowlabs 2023-01-12 19:20:22 +00:00
parent f0dd9711ed
commit 2edfb1a21f
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -267,7 +267,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", str(prediction_mask.numpy().tolist()).replace("], [", "],\n["))
# print("DEBUG:plot_predictions INFER", str(prediction_mask.numpy().tolist()).replace("], [", "],\n["))
plot_samples_matplotlib(
filepath.replace("$$", str(i)),