wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-16 11:22:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Correct name depth to space instead of depth to scale

Browse Source
This commit is contained in:
Somshubra Majumdar
2017-02-27 23:04:54 -06:00
parent 2e466d07d7
commit c40c680767
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/backend/tensorflow_backend.py
+1 -1
View File
@@ -103,7 +103,7 @@ def extract_image_patches(X, ksizes, ssizes, border_mode="same", dim_ordering="t
def depth_to_space(input, scale):
''' Uses phase shift algorithm to convert channels/depth for spatial resolution '''
assert K.image_dim_ordering() == 'tf', 'depth_to_scale backend function can only be used with "tf" dim ' \
assert K.image_dim_ordering() == 'tf', 'depth_to_space backend function can only be used with "tf" dim ' \
'ordering when using tensorflow backend'
return tf.depth_to_space(input, scale)
keras_contrib/backend/theano_backend.py
+1 -1
View File
@@ -121,7 +121,7 @@ def extract_image_patches(X, ksizes, strides, border_mode="valid", dim_ordering=
def depth_to_space(input, scale):
''' Uses phase shift algorithm to convert channels/depth for spatial resolution '''
assert K.image_dim_ordering() == 'th', 'depth_to_scale backend function can only be used with "th" dim ' \
assert K.image_dim_ordering() == 'th', 'depth_to_space backend function can only be used with "th" dim ' \
'ordering when using theano backend'
b, k, row, col = input.shape