fix sub_pixel_upscaling test

This commit is contained in:
Michael Oliver
2017-06-09 09:08:17 -07:00
committed by GitHub
parent 0182528dd2
commit 0f78409087
@@ -156,9 +156,11 @@ def test_sub_pixel_upscaling():
num_samples = 2
num_row = 16
num_col = 16
input_dtype = K.floatx()
for scale_factor in [2, 3, 4]:
input_data = np.random.random((num_samples, 4 * (scale_factor ** 2), num_row, num_col))
input_data = input_data.astype(input_dtype)
if K.image_data_format() == 'channels_last':
input_data = input_data.transpose((0, 2, 3, 1))