From 0e3de8f5fc9edd9b2394e7b867c2dae95afd40c6 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 10 Jan 2023 19:19:30 +0000 Subject: [PATCH] dlr: fix predictions --- aimodel/src/deeplabv3_plus_test_rainfall.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index 1af1521..5bf115f 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -252,10 +252,10 @@ def plot_predictions(filepath, input_items, colormap, model): for input_tensor in input_items: prediction_mask = infer(image_tensor=input_tensor, model=model) prediction_colormap = decode_segmentation_masks(prediction_mask, colormap, 20) - overlay = get_overlay(input_tensor, prediction_colormap) + plot_samples_matplotlib( filepath, - [input_tensor, overlay, prediction_colormap], + [input_tensor, prediction_colormap], figsize=(18, 14) )