Source code for skgtimage.utils.quickshift
import numpy as np
import time
import skimage; from skimage import segmentation
[docs]def quickshift(image,ratio,mc=False,roi=None,verbose=False):
"""
Apply quickshift segmentation to input image (within region of interest)
:param image: input image
:param ratio: ratio parameter considered in scikit-image quickshift
:param mc: whether image is multi-component or not (color in our case)
:param verbose:
:return: labelled image (numpy array), where each label corresponds to a specific value
"""
input_image=image
if mc is False:
input_image=np.dstack((image,image,image))
t0=time.clock()
label=skimage.segmentation.quickshift(input_image,ratio=ratio)
t1 = time.clock()
if verbose:
print("Cpu time (sec): " , t1-t0)
if roi is not None:
label+=1
label=np.ma.array(label, mask=np.logical_not(roi)).filled(0)
return label