From 8d932757b584ae4eecf44a97abf77122229797e4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 9 Mar 2023 19:44:39 +0000 Subject: [PATCH] dlr: variable name typo --- 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 24d4f5a..edd17e5 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -307,7 +307,7 @@ def plot_predictions(filepath, input_items, colormap, model): i = 0 for input_pair in input_items: prediction_mask = infer(image_tensor=input_pair[0], model=model) - prediction_mask_argmax = tf.argmax(predictions, axis=2) + prediction_mask_argmax = tf.argmax(prediction_mask, axis=2) # label_colourmap = decode_segmentation_masks(input_pair[1], colormap, 2) prediction_colormap = decode_segmentation_masks(prediction_mask_argmax, colormap, 2)