From 69b5ae8838be84b9e824e9f083754edff4cf2634 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 23 Feb 2023 17:24:30 +0000 Subject: [PATCH] dlr eo: this should fix it --- 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 a32d6d1..d4520d3 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -130,7 +130,7 @@ if PATH_CHECKPOINT is None: x = DilatedSpatialPyramidPooling(x) input_a = tf.keras.layers.UpSampling2D( - size=(image_size // 4 // x.shape[1], image_size // 4 // x.shape[2]), # <--- UPSAMPLE after pyramid + size=(image_size // 4 // x.shape[1] * 2, image_size // 4 // x.shape[2] * 2), # <--- UPSAMPLE after pyramid interpolation="bilinear", )(x) input_b = resnet50.get_layer("conv2_block3_2_relu").output