From 3787155665709467d80ff8b23b6bace47f9f06b4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 11 Jan 2023 17:39:14 +0000 Subject: [PATCH] dlr: we can't plot the input tensor because it has 8 channels rather than 1 or 3 --- aimodel/src/deeplabv3_plus_test_rainfall.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index 294cab4..c247154 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -262,7 +262,10 @@ def plot_predictions(filepath, input_items, colormap, model): plot_samples_matplotlib( filepath, - [input_tensor, prediction_colormap], + [ + # input_tensor, + prediction_colormap + ], figsize=(18, 14) )