From cb5e0b2369ba633f9585b495f21e6275a90fc16d Mon Sep 17 00:00:00 2001 From: Achilles Rasquinha Date: Sun, 9 Feb 2020 19:33:35 -0600 Subject: [PATCH 1/3] Fixed tf image resize in pspnet_2 --- keras_segmentation/models/_pspnet_2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_segmentation/models/_pspnet_2.py b/keras_segmentation/models/_pspnet_2.py index 844b7c74c..be0a232f7 100644 --- a/keras_segmentation/models/_pspnet_2.py +++ b/keras_segmentation/models/_pspnet_2.py @@ -34,7 +34,7 @@ def build(self, input_shape): def call(self, inputs, **kwargs): new_height, new_width = self.new_size - resized = tf.image.resize_images(inputs, [new_height, new_width], + resized = tf.image.resize(inputs, [new_height, new_width], align_corners=True) return resized From ddcf6af35bd6754e6ed681786543b4369f486bef Mon Sep 17 00:00:00 2001 From: Achilles Rasquinha Date: Sun, 9 Feb 2020 19:36:37 -0600 Subject: [PATCH 2/3] Fixed tf image resize in pspnet_2 --- keras_segmentation/models/_pspnet_2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras_segmentation/models/_pspnet_2.py b/keras_segmentation/models/_pspnet_2.py index be0a232f7..3073fe0f8 100644 --- a/keras_segmentation/models/_pspnet_2.py +++ b/keras_segmentation/models/_pspnet_2.py @@ -34,8 +34,8 @@ def build(self, input_shape): def call(self, inputs, **kwargs): new_height, new_width = self.new_size - resized = tf.image.resize(inputs, [new_height, new_width], - align_corners=True) + resized = tf.image.resize(inputs, [new_height, new_width]) + return resized def compute_output_shape(self, input_shape): From cd0e4305893d92ff6d3a98801dfb5b1c2cc00cad Mon Sep 17 00:00:00 2001 From: Achilles Rasquinha Date: Tue, 18 Feb 2020 10:40:05 -0600 Subject: [PATCH 3/3] Support for all tensorflow versions --- keras_segmentation/models/_pspnet_2.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/keras_segmentation/models/_pspnet_2.py b/keras_segmentation/models/_pspnet_2.py index 3073fe0f8..4265ef5d4 100644 --- a/keras_segmentation/models/_pspnet_2.py +++ b/keras_segmentation/models/_pspnet_2.py @@ -34,7 +34,9 @@ def build(self, input_shape): def call(self, inputs, **kwargs): new_height, new_width = self.new_size - resized = tf.image.resize(inputs, [new_height, new_width]) + + resized = tf.compat.v1.image.resize(inputs, [new_height, new_width], + align_corners=True) return resized