Skip to content

Commit d8854fb

Browse files
committed
2 parents 692473b + 045880b commit d8854fb

File tree

1 file changed

+2
-2
lines changed
  • machine-learning/hog-feature-extraction

1 file changed

+2
-2
lines changed

machine-learning/hog-feature-extraction/hog.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
#creating hog features
2121
fd, hog_image = hog(resized_img, orientations=9, pixels_per_cell=(8, 8),
22-
cells_per_block=(2, 2), visualize=True, multichannel=True)
22+
cells_per_block=(2, 2), visualize=True, channel_axis=-1)
2323
print(fd.shape)
2424
print(hog_image.shape)
2525
plt.axis("off")
@@ -28,4 +28,4 @@
2828

2929
# save the images
3030
plt.imsave("resized_img.jpg", resized_img)
31-
plt.imsave("hog_image.jpg", hog_image, cmap="gray")
31+
plt.imsave("hog_image.jpg", hog_image, cmap="gray")

0 commit comments

Comments
 (0)