o :ήc)@sddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z m Z gdZd Zd d e jDZd d ZddZ  ddddddZ   dddZdddZdS)N)_supported_float_typewarn) img_as_float) rgb_colors)gray2rgbrgb2hsvhsv2rgb) color_dict label2rgbDEFAULT_COLORS) redblueyellowmagentagreenindigo darkorangecyanpink yellowgreencCs i|] \}}t|tr||qS) isinstancetuple).0kvrr?/tmp/pip-target-vg8gfxp4/lib/python/skimage/color/colorlabel.py s  rcCs$t|tr t|}t|ddS)aAReturn RGB color as (1, 3) array. This RGB array gets multiplied by masked regions of an RGB image, which are partially flattened by masking (i.e. dimensions 2D + RGB -> 1D + RGB). Parameters ---------- color : str or array Color name in `color_dict` or RGB float values between [0, 1]. N)rstrr nparray)colorrrr _rgb_vectors r%c Cs|durd}t|}tj|dd\}}||j|k}t|dkr6|d}|||kd7<d||j|k<n|d7}t|}t|g|}||fS)zReturn `unique_labels` and `color_cycle` for label array and color list. Colors are cycled for normal labels, but the background color should only be used for the background. NrrrT)return_inverserr)r%r"uniqueflatlen itertoolscyclechain) labelcolorsbg_labelbg_color unique_labels mapped_labelsbg_label_rank_list bg_label_rank color_cyclerrr_match_label_with_color's  r7333333?r&overlay) saturation channel_axisc Csh|dur tj|| dd}|dkrt||||||||} n|dkr(t||||} ntdtj| d| dS)aReturn an RGB image where color-coded labels are painted over the image. Parameters ---------- label : ndarray Integer array of labels with the same shape as `image`. image : ndarray, optional Image used as underlay for labels. It should have the same shape as `labels`, optionally with an additional RGB (channels) axis. If `image` is an RGB image, it is converted to grayscale before coloring. colors : list, optional List of colors. If the number of labels exceeds the number of colors, then the colors are cycled. alpha : float [0, 1], optional Opacity of colorized labels. Ignored if image is `None`. bg_label : int, optional Label that's treated as the background. If `bg_label` is specified, `bg_color` is `None`, and `kind` is `overlay`, background is not painted by any colors. bg_color : str or array, optional Background color. Must be a name in `color_dict` or RGB float values between [0, 1]. image_alpha : float [0, 1], optional Opacity of the image. kind : string, one of {'overlay', 'avg'} The kind of color image desired. 'overlay' cycles over defined colors and overlays the colored labels over the original image. 'avg' replaces each labeled segment with its average color, for a stained-class or pastel painting appearance. saturation : float [0, 1], optional Parameter to control the saturation applied to the original image between fully saturated (original RGB, `saturation=1`) and fully unsaturated (grayscale, `saturation=0`). Only applies when `kind='overlay'`. channel_axis : int, optional This parameter indicates which axis of the output array will correspond to channels. If `image` is provided, this must also match the axis of `image` that corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- result : ndarray of float, same shape as `image` The result of blending a cycling colormap (`colors`) for each distinct value in `label` with the image, at a certain alpha value. Nr:)source destinationr9avgz)`kind` must be either 'overlay' or 'avg'.)r"moveaxis_label2rgb_overlay_label2rgb_avg ValueError) r.imager/alphar0r1 image_alphakindr;r<rgbrrrr Js3 r cCs(d|kr dksntd||durt}dd|D}|dur1tj|jdtjd}d}nk|jd|j|jksD|j|jdkrHtd |j|jdkr[|jd d kr[td |dkretd t |j }t |j |dd}|j|jkrt |} | d|9<t| }n |j|jkrt|}||d|}t||} | dkr|| }|| 8}tt|} | tkrtj} | | }t||||\} } t| dkr|Stt| d}tddt|| D}|}| |_||||d|}d| vo|du}|r|||k|||k<|S)aReturn an RGB image where color-coded labels are painted over the image. Parameters ---------- label : ndarray Integer array of labels with the same shape as `image`. image : ndarray, optional Image used as underlay for labels. It should have the same shape as `labels`, optionally with an additional RGB (channels) axis. If `image` is an RGB image, it is converted to grayscale before coloring. colors : list, optional List of colors. If the number of labels exceeds the number of colors, then the colors are cycled. alpha : float [0, 1], optional Opacity of colorized labels. Ignored if image is `None`. bg_label : int, optional Label that's treated as the background. If `bg_label` is specified and `bg_color` is `None`, background is not painted by any colors. bg_color : str or array, optional Background color. Must be a name in `color_dict` or RGB float values between [0, 1]. image_alpha : float [0, 1], optional Opacity of the image. saturation : float [0, 1], optional Parameter to control the saturation applied to the original image between fully saturated (original RGB, `saturation=1`) and fully unsaturated (grayscale, `saturation=0`). Returns ------- result : ndarray of float, same shape as `image` The result of blending a cycling colormap (`colors`) for each distinct value in `label` with the image, at a certain alpha value. rrz(saturation must be in range [0, 1], got NcSsg|]}t|qSr)r%)rcrrr z&_label2rgb_overlay..r dtypez*`image` and `label` must be the same shaper:r z0`image` must be RGB (image.shape[-1] must be 3).z1Negative intensities in `image` are not supportedF)copy).rcSsg|]\}}|qSrr)rirIrrrrJrK)rr r"zerosshapefloat64ndimrCminrrNrastyper r rmin_scalar_typeintmaxbooluint8r7r*rangestackzipr))r.rDr/rEr0r1rFr; float_dtypehsvoffsetnew_typemapped_labels_flatr6 dense_labelslabel_to_colorr3resultremove_backgroundrrrrAsd%       rAc Cstj|jd|jd}t|}||k}|r(|||k}||k}|||<|D]}||k}||jdd} | ||<q*|S)aCVisualise each segment in `label_field` with its mean color in `image`. Parameters ---------- label_field : ndarray of int A segmentation of an image. image : array, shape ``label_field.shape + (3,)`` A color image of the same spatial shape as `label_field`. bg_label : int, optional A value in `label_field` to be treated as background. bg_color : 3-tuple of int, optional The color for the background label Returns ------- out : ndarray, same shape and type as `image` The output visualization. rLrMr)axis)r"rQrRrNr(anynonzeromean) label_fieldrDr0r1outlabelsbgmaskr.r$rrrrBs     rB)NNr8rr&rr9)NNr8r:Nrr)rr&)r+numpyr" _shared.utilsrrutilrr colorconvrr r __all__r __dict__itemsr r%r7r rArBrrrrs*  # ? j