I noticed that cy+cy and cx+cx was repeated in the inner-most loop as well

This commit is contained in:
Ivo Flipse
2015-06-18 22:01:44 +02:00
parent ded652c2ab
commit bea9886a12
+3 -1
View File
@@ -128,11 +128,13 @@ def hog(image, orientations=9, pixels_per_cell=(8, 8),
dx_arr = radius * np.cos(orientations_arr / orientations_arr * np.pi)
dy_arr = radius * np.sin(orientations_arr / orientations_arr * np.pi)
cc = cy + cy
cr = cx + cx
hog_image = np.zeros((sy, sx), dtype=float)
for x in range(n_cellsx):
for y in range(n_cellsy):
for o, dx, dy in zip(orientations_arr, dx_arr, dy_arr):
centre = tuple([y * cy + cy // 2, x * cx + cx // 2])
centre = tuple([y * cc // 2, x * cr // 2])
rr, cc = draw.line(int(centre[0] - dx),
int(centre[1] + dy),