contours.py 文件源码

def count_neighbors(binary):
    convolve = 3**binary.ndim * ndimage.uniform_filter(binary.astype(np.float)) # 3x3 sum
    neighbors = np.copy(binary.astype(np.uint8))
    binary = binary.astype(np.bool)
    neighbors[binary] = convolve[binary] - 1
    return neighbors