commit e4e204d921b653bcaa3af0b3cf07899a2155689a
author Lun Dong <lundong@google.com> Sat Jul 30 16:48:29 2022 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Sat Jul 30 16:48:29 2022 +0000
tree 6693e58040ec1d2495b104c5504bc99575f3a432
parent 5e0c3f8213466e38ed8e85f5a3bea5d514ceb211
parent cedef3ddb88d43209e978fe0618760206d817861

Merge "Bug fix on resizing input image"

build_tools/gen_mlmodel_input.py

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)