Esempio n. 1
0
def run_naive(img, mask, mu, sigma, niters=1):
    S = Segmentation(img.get_data(), mask=mask, mu=mu, sigma=sigma, beta=0)
    S.ve_step()
    return S
def run_vem(img, mask, mu, sigma):
    S = Segmentation(img.get_data(), mask=mask, mu=mu, sigma=sigma,
                 ngb_size=NGB_SIZE, beta=BETA)
    for it in range(NITERS):
        S.ve_step()
    return S