From 864dfa802d48282c9d28193b0b6a0ebe6bb22749 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 12 Jan 2023 18:03:06 +0000 Subject: [PATCH] dlr/predict 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 3a2eaba..c3080c5 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -247,7 +247,7 @@ def get_overlay(image, coloured_mask): def plot_samples_matplotlib(filepath, display_list, figsize=(5, 3)): for i in range(len(display_list)): - plt.subplot(1, len(display_list), i) + plt.subplot(1, len(display_list), i+1) if display_list[i].shape[-1] == 3: plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i])) else: