distance_plgs.py 文件源码

def run(self, ips, snap, img, para = None):
        img[:] = snap
        dist = -ndimg.distance_transform_edt(snap)
        pts = find_maximum(dist, para['tor'], False)
        buf = np.zeros(ips.size, dtype=np.uint16)
        buf[pts[:,0], pts[:,1]] = 1
        markers, n = ndimg.label(buf, np.ones((3,3)))
        line = watershed(dist, markers, line=True, conn=para['con']+1)
        img[line==0] = 0