From ad52ae9241858a966512a07066dcd0820e1dcd4c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 9 Mar 2023 19:13:25 +0000 Subject: [PATCH] dir: also plot inputs --- aimodel/src/deeplabv3_plus_test_rainfall.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index 3f190cd..b996766 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -313,6 +313,8 @@ def plot_predictions(filepath, input_items, colormap, model): filepath.replace("$$", str(i)), [ # input_tensor, + tf.math.reduce_max(input_pair[0][:,:,:-1], axis=-1), # rainfall only + input_pair[0][:,:,-1], # heightmap input_pair[1], #label_colourmap, prediction_mask[:,:,1], prediction_colormap