Mercurial > repos > imgteam > voronoi_tesselation
view voronoi_tessellation.py @ 2:412d7478ec92 draft
planemo upload for repository https://github.com/BMCV/galaxy-image-analysis/tree/master/tools/voronoi_tesselation commit c86a1b93cb7732f7331a981d13465653cc1a2790
author | imgteam |
---|---|
date | Wed, 24 Apr 2024 08:13:41 +0000 |
parents | 8b74843c136e |
children |
line wrap: on
line source
import argparse import giatools.io import numpy as np import scipy.ndimage as ndi import skimage.io from skimage.segmentation import watershed if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('input') parser.add_argument('output') args = parser.parse_args() im = giatools.io.imread(args.input) im = im.squeeze() assert im.ndim == 2 distances = np.full(im.shape, np.inf) for label in np.unique(im): if label == 0: continue label_distances = ndi.distance_transform_edt(im != label) distances = np.min((distances, label_distances), axis=0) result = watershed( image=distances, markers=im, ) skimage.io.imsave(args.output, result)