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
