Merge pull request #822 from chrishkchris/lstm11122020
Fix new lstm error
diff --git a/python/singa/image_tool.py b/python/singa/image_tool.py
index 1011db8..346b29a 100644
--- a/python/singa/image_tool.py
+++ b/python/singa/image_tool.py
@@ -139,9 +139,9 @@
return resize(img, small_size)
-def resize_by_hw(img, (height, width)):
+def resize_by_hw(img, size):
'''Resize the image to be the given width and height'''
- new_img = img.resize((width, height))
+ new_img = img.resize((size[1], size[0]))
return new_img
