wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-09-12 12:32:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

update depth-to-space

Browse Source
This commit is contained in:
farizrahman4u
2017-03-19 08:00:36 +05:30
parent 8ce313aaff
commit 4c8a5a939a
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/backend/theano_backend.py
+6 -4
View File
@@ -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