From 4c8a5a939ac1a98f02e0cf4fd076b34dcbc305ae Mon Sep 17 00:00:00 2001 From: farizrahman4u Date: Sun, 19 Mar 2017 08:00:36 +0530 Subject: [PATCH] update depth-to-space --- keras_contrib/backend/theano_backend.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/keras_contrib/backend/theano_backend.py b/keras_contrib/backend/theano_backend.py index 6e7be0d..7080a8d 100644 --- a/keras_contrib/backend/theano_backend.py +++ b/keras_contrib/backend/theano_backend.py @@ -184,10 +184,12 @@ def extract_image_patches(X, ksizes, strides, padding="valid", data_format="chan return patches -def depth_to_space(input, scale): +def depth_to_space(input, scale, data_format=None): ''' Uses phase shift algorithm to convert channels/depth for spatial resolution ''' - - input = _preprocess_conv2d_input(input, image_data_format()) + if data_format is None: + data_format = image_data_format() + data_format = data_format.lower() + input = _preprocess_conv2d_input(input, data_format) b, k, row, col = input.shape output_shape = (b, k // (scale ** 2), row * scale, col * scale) @@ -198,7 +200,7 @@ def depth_to_space(input, scale): for y, x in itertools.product(range(scale), repeat=2): out = T.inc_subtensor(out[:, :, y::r, x::r], input[:, r * y + x:: r * r, :, :]) - out = _postprocess_conv2d_output(out, input, None, None, None, image_data_format()) + out = _postprocess_conv2d_output(out, input, None, None, None, data_format) return out