Fixed 2D compatibility for real this time
@@ -86,7 +86,7 @@ def walk(img, st_point, maxsteps, stride=1):
z, y, x = st_point
directions = 6
elif ndim == 2:
- y, x = st_point[0:2]
+ y, x = st_point[1:3]
z = 0
directions = 4
img = np.array([img])
