Exemplo n.º 1
0
def checkAlgorithmSameObjects(image):
    # check if same objects after and before pruning
    skel = Skeleton(image)
    skel.setPrunedSkeletonOutput()
    label_img, countObjectsn = ndimage.measurements.label(skel.outputStack, structure=np.ones((3, 3, 3), dtype=np.uint8))
    label_img, countObjects = ndimage.measurements.label(image, structure=np.ones((3, 3, 3), dtype=np.uint8))
    assert (countObjectsn == countObjects) or np.sum(label_img == 2) == 1