summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--python/caffe/io.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/io.py b/python/caffe/io.py
index fc962660..0cad7211 100644
--- a/python/caffe/io.py
+++ b/python/caffe/io.py
@@ -329,7 +329,7 @@ def resize_image(im, new_dims, interp_order=1):
return ret
else:
# ndimage interpolates anything but more slowly.
- scale = tuple(np.array(new_dims) / np.array(im.shape[:2]))
+ scale = tuple(np.array(new_dims, dtype=float) / np.array(im.shape[:2]))
resized_im = zoom(im, scale + (1,), order=interp_order)
return resized_im.astype(np.float32)