Add ndim argument in utility method

This commit is contained in:
Steven Silvester
2014-09-21 06:10:40 -05:00
parent ded8fa30b9
commit 2db19c624d
+1 -1
View File
@@ -125,7 +125,7 @@ def plot_matches(ax, image1, image2, keypoints1, keypoints2, matches,
def _prepare_grayscale_input_2D(image):
image = np.squeeze(image)
assert_nD(image)
assert_nD(image, 2)
return img_as_float(image)