def mid_axis(img): dis = ndimg.distance_transform_edt(img) idx = np.argsort(dis.flat).astype(np.int32) medial_axis(dis, idx, lut) return dis