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

image_segmentation.py pep8

Browse Source
This commit is contained in:
Andrew Hundt
2017-12-19 15:53:48 -05:00
parent 3508fdd714
commit d54f71a7d2
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/preprocessing/image_segmentation.py
+4 -4
View File
@@ -37,11 +37,11 @@ def pair_center_crop(x, y, center_crop_size, data_format, **kwargs):
h_start, h_end = centerh - lh, centerh + rh
w_start, w_end = centerw - lw, centerw + rw
if data_format == 'channels_first':
return x[:, h_start:h_end, w_start:w_end], \
y[:, h_start:h_end, w_start:w_end]
return (x[:, h_start:h_end, w_start:w_end],
y[:, h_start:h_end, w_start:w_end])
elif data_format == 'channels_last':
return x[h_start:h_end, w_start:w_end, :], \
y[h_start:h_end, w_start:w_end, :]
return (x[h_start:h_end, w_start:w_end, :],
y[h_start:h_end, w_start:w_end, :])
def random_crop(x, random_crop_size, data_format, sync_seed=None, **kwargs):