Merge pull request #3070 from mohomran/resize_nd-image_bug

bug in caffe.io.resize_image when applied to images with 2 or >3 channels
This commit is contained in:
Jeff Donahue 2015-09-16 11:24:27 -07:00
Родитель e4baef28d6 b7f9cba875
Коммит 71e05876f6
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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)