diff --git a/build_tools/gen_mlmodel_input.py b/build_tools/gen_mlmodel_input.py
index 213d508..fb87b3a 100755
--- a/build_tools/gen_mlmodel_input.py
+++ b/build_tools/gen_mlmodel_input.py
@@ -67,7 +67,7 @@
         input = input[:np.prod(input_shape)].reshape(np.prod(input_shape))
     else:
         resized_img = Image.open(input_name).resize(
-            (input_shape[1], input_shape[2]))
+            (input_shape[2], input_shape[1]))
         input = np.array(resized_img).reshape(np.prod(input_shape))
         if not is_quant:
             low = np.min(float_input_range)