Bug fix on resizing input image
The input tuple for resizing an image is (width, height). It should
correspnds to (input_shape[2], input_shape[1]).
In mobilenet/mnist, input width = height, so this issue was not captured
before.
Change-Id: I6cf61bca3e8e2f5c163c8e5dec414cf674105324
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)
