From 808e53c7ae53ad80e1c8f8ed77b34f1b38b569df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pablo=20M=C3=A1rquez=20Neila?= Date: Mon, 18 Aug 2014 02:12:27 +0200 Subject: [PATCH] Code valid for all image dtypes --- skimage/feature/_hog.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skimage/feature/_hog.py b/skimage/feature/_hog.py index f9608c9d..2c6dbf0a 100644 --- a/skimage/feature/_hog.py +++ b/skimage/feature/_hog.py @@ -80,11 +80,11 @@ def hog(image, orientations=9, pixels_per_cell=(8, 8), # to avoid problems with subtracting unsigned numbers in np.diff() image = image.astype('float') - gx = np.empty_like(image) + gx = np.empty(image.shape) gx[:, 0] = 0 gx[:, -1] = 0 gx[:, 1:-1] = image[:, 2:] - image[:, :-2] - gy = np.empty_like(image) + gy = np.empty(image.shape) gy[0, :] = 0 gy[-1, :] = 0 gy[1:-1, :] = image[2:, :] - image[:-2, :]