From 2d0507ea3839a7f7bc1e95a28e8b10e4896d1452 Mon Sep 17 00:00:00 2001 From: ondrej Date: Mon, 28 Aug 2023 17:26:24 +0200 Subject: [PATCH] fix FCN: wrong argument name for UpSampling2D --- src/architectures.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/architectures.py b/src/architectures.py index c8e221fc..407e3e25 100644 --- a/src/architectures.py +++ b/src/architectures.py @@ -1230,7 +1230,7 @@ def instantiate_layers(self): ) if self.variant == '32s': - self.final_upsample = UpSampling2D(pool_size=(32, 32), + self.final_upsample = UpSampling2D(size=(32, 32), name='upsampling_final') return 0 elif self.variant == '16s': @@ -1243,7 +1243,7 @@ def instantiate_layers(self): self.adds = [] for i in range(5, stop_level, -1): self.upsamples.append( - UpSampling2D(pool_size=(2, 2), + UpSampling2D(size=(2, 2), name=f'upsampling_{i}_to_{i - 1}')) self.classifiers.append(Conv2D( self.nr_classes, @@ -1255,8 +1255,8 @@ def instantiate_layers(self): )) self.adds.append(Concatenate(axis=3, name=f'concat_{i}_to_{i - 1}')) - self.final_upsample = UpSampling2D(pool_size=(2 ** stop_level, - 2 ** stop_level), + self.final_upsample = UpSampling2D(size=(2 ** stop_level, + 2 ** stop_level), name='upsampling_final') def get_config(self):