Corrected testing code

This commit is contained in:
Vighnesh Birodkar
2015-06-07 00:40:05 +05:30
parent 2cc4066097
commit 66caaa5b9f
+1 -1
View File
@@ -5,7 +5,7 @@ from numpy import testing
def energy(img): def energy(img):
if(img.ndim == 3): if(img.ndim == 3):
return np.ascontiguousarray(img[:, :, 0]) img = np.ascontiguousarray(img[:, :, 0])
return (1 - img) return (1 - img)