quick fix

keras-team · Nov 27, 2018 · f5a0081 · f5a0081
1 parent 1dfa3d5
commit f5a0081
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autokeras/utils.py b/autokeras/utils.py
@@ -207,7 +207,7 @@ def compute_image_resize_params(data):
     if data is None or len(data.shape) == 0:
         return []
 
-    if len(data.shape) == len(data[0].shape) + 1:
+    if len(data.shape) == len(data[0].shape) + 1 and np.prod(data[0].shape[:-1]) <= Constant.MAX_IMAGE_SIZE:
         return data[0].shape
 
     data_shapes = []