From 1b5bb14d8fe2d38772a163080c804ddb0cb5fe90 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 12 Jan 2023 18:35:29 +0000 Subject: [PATCH] dlr: debug --- aimodel/src/deeplabv3_plus_test_rainfall.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index f0880c5..8b35dfb 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -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)),