diff --git a/machine-learning/hog-feature-extraction/hog.ipynb b/machine-learning/hog-feature-extraction/hog.ipynb index 0b6c2852..8158380a 100644 --- a/machine-learning/hog-feature-extraction/hog.ipynb +++ b/machine-learning/hog-feature-extraction/hog.ipynb @@ -69,7 +69,7 @@ "source": [ "#creating hog features\n", "fd, hog_image = hog(resized_img, orientations=9, pixels_per_cell=(8, 8),\n", - " \tcells_per_block=(2, 2), visualize=True, multichannel=True)\n", + " \tcells_per_block=(2, 2), visualize=True, channel_axis=2)\n", "print(fd.shape)\n", "plt.axis(\"off\")\n", "plt.imshow(hog_image, cmap=\"gray\")" @@ -94,4 +94,4 @@ "source": [] } ] -} \ No newline at end of file +}