commit | d89f2f89adcab11d13bd1a8fbca0bdb4a270e077 | [log] [tgz] |
---|---|---|
author | Chris Yeung <chrishkchris@yahoo.com.hk> | Fri Dec 11 13:02:40 2020 +0800 |
committer | Chris Yeung <chrishkchris@yahoo.com.hk> | Fri Dec 11 13:02:40 2020 +0800 |
tree | 579ed29bddd5771ae3d7c41983cf5dcd50decd3e | |
parent | 1862c726389b5b254380ee99357c7a12ea658f28 [diff] |
fix lstm error
diff --git a/python/singa/image_tool.py b/python/singa/image_tool.py index 1011db8..0322316 100644 --- a/python/singa/image_tool.py +++ b/python/singa/image_tool.py
@@ -330,7 +330,7 @@ small_sizes = get_list_sample(size_list, num_case) for small_size in small_sizes: - new_img = resize_by_hw(img, small_size) + new_img = resize_by_hw(img, (small_size[0], small_size[1])) new_imgs.append(new_img) if inplace: self.imgs = new_imgs