Merge pull request #1115 from dalonsoa/bug_zoom_multi

Fix inconsistent output in zoom_multi
tensorlayer · Feb 1, 2021 · a9f77e4 · a9f77e4
2 parents 066c09b + 34cf1df
commit a9f77e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorlayer/prepro.py b/tensorlayer/prepro.py
@@ -1527,7 +1527,7 @@ def zoom_multi(x, zoom_range=(0.9, 1.1), flags=None, border_mode='constant'):
         h, w = x.shape[0], x.shape[1]
         transform_matrix = transform_matrix_offset_center(zoom_matrix, h, w)
         results.append(affine_transform_cv2(x, transform_matrix, flags=flags, border_mode=border_mode))
-    return results
+    return np.asarray(results)
 
 
 # image = tf.image.random_brightness(image, max_delta=32. / 255.)