o e=@sdZddlmZmZmZddlZddlZddlm Z ddl m Z ddl m Z dd l mZdd l mZd d d d de jfddZddZddZddZddZddZddZddZGdd d e jZdS)!zn Augmenters that apply artistic image filters. List of augmenters: * :class:`Cartoon` Added in 0.4.0. )print_functiondivisionabsolute_importN)_normalize_cv2_input_arr_)meta)color)dtypes) parameters?g@Tc CsXtj|dggddd|jdkr|jddks Jd|jfttt|d}t|d }t|d }t|jd dd k}t||}t |}|rXtd |} td |} n td|} td|} |d krk|}n t j t || | |d|rt |||} nt|||} | } | dkdtj} |rt| dddd} t| dddd} tt|| ||S)aE Convert the style of an image to a more cartoonish one. This function was primarily designed for images with a size of ``200`` to ``800`` pixels. Smaller or larger images may cause issues. Note that the quality of the results can currently not compete with learned style transfer, let alone human-made images. A lack of detected edges or also too many detected edges are probably the most significant drawbacks. This method is loosely based on the one proposed in https://stackoverflow.com/a/11614479/3760780 Added in 0.4.0. **Supported dtypes**: * ``uint8``: yes; fully tested * ``uint16``: no * ``uint32``: no * ``uint64``: no * ``int8``: no * ``int16``: no * ``int32``: no * ``int64``: no * ``float16``: no * ``float32``: no * ``float64``: no * ``float128``: no * ``bool``: no Parameters ---------- image : ndarray A ``(H,W,3) uint8`` image array. blur_ksize : int, optional Kernel size of the median blur filter applied initially to the input image. Expected to be an odd value and ``>=0``. If an even value, thn automatically increased to an odd one. If ``<=1``, no blur will be applied. segmentation_size : float, optional Size multiplier to decrease/increase the base size of the initial mean-shift segmentation of the image. Expected to be ``>=0``. Note that the base size is increased by roughly a factor of two for images with height and/or width ``>=400``. edge_prevalence : float, optional Multiplier for the prevalance of edges. Higher values lead to more edges. Note that the default value of ``1.0`` is already fairly conservative, so there is limit effect from lowerin it further. saturation : float, optional Multiplier for the saturation. Set to ``1.0`` to not change the image's saturation. suppress_edges : bool, optional Whether to run edge suppression to remove blobs containing too many or too few edge pixels. from_colorspace : str, optional The source colorspace. Use one of ``imgaug.augmenters.color.CSPACE_*``. Defaults to ``RGB``. Returns ------- ndarray Image in cartoonish style. uint8)boolZuint16Zuint32Zuint64Zuint128Zuint256Zint8Zint16int32Zint64Zint128Zint256Zfloat16float32Zfloat64Zfloat96Zfloat128Zfloat256N)allowedZ disallowedZ augmenterr r z.Expected to get a (H,W,C) image, got shape %s.rri ()spsrdstdF)inverseT)iadtZ gate_dtypesndimshapemaxintnpround _blur_medianZ zeros_likecv2ZpyrMeanShiftFilteringr_find_edges_canny_find_edges_laplacianastyper_suppress_edge_blobs _saturate _blend_edges) image blur_ksizesegmentation_size saturationedge_prevalenceZsuppress_edgesfrom_colorspaceZis_small_imageZ image_segZspatial_window_radiusZcolor_window_radiusZ edges_rawedgesr6JD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmenters/artistic.pystylize_cartoonsZK          r8cCsLtjt|tj|d}|d}ttdd|d}tt |||}|S)NZ to_colorspacer4.rr) colorlibchange_colorspace_r%copy CSPACE_GRAYminr$r(ZCannyr)r/edge_multiplierr4 image_graythreshr5r6r6r7r)s r)cCstjt|tj|d}|d}tt|dtj}t |}|d}t |t t dd|d}t |d||}t t|d d dtj}t|d }t|d }|S) Nr9r:o@r Zrcrrrr 2)r=r>r%r?r@r(Z LaplacianrZCV_64FabsZ percentilerAr$clipr&r+rr' _threshold)r/rBr4rCZedges_fZvmaxZ edges_uint8r6r6r7r*s     r*cCs0|ddkr |d7}|dkr|Stt||S)Nr rr)r(Z medianBlurr)r/Zksizer6r6r7r's r'cCs||k}t|}d||<|S)Nr)r%r?)r/rDmaskresultr6r6r7rKs rKcCsVtj||ftjd}tt|dd|}|r||k}n||k}t|}d||<|S)N)ZdtyperEr)r%Zonesrr(Zfilter2Drr?)r5sizerDrZkernelcountsrLr6r6r7r,s  r,cCst|}tj|dddr|Stj|tj|d}|dddddf}t|tj|ddtj }||dddddf<tj||tjd}|S)Nr g{Gz?)Zatolr9rrr) r%r?iscloser=r>Z CSPACE_HSVrJr+rr)r/factorr4ZhsvsatZ image_satr6r6r7r-s "r-cCsBd|d}t|dtjfd}tt||ddtjS)Nr rE.)rrr r)r%ZtileZnewaxisrJr&r+r)r/Z image_edgesr6r6r7r.s  r.c sNeZdZdZddddejddddf fdd Zd d Zd d ZddZ Z S)CartoonaConvert the style of images to a more cartoonish one. This augmenter was primarily designed for images with a size of ``200`` to ``800`` pixels. Smaller or larger images may cause issues. Note that the quality of the results can currently not compete with learned style transfer, let alone human-made images. A lack of detected edges or also too many detected edges are probably the most significant drawbacks. Added in 0.4.0. **Supported dtypes**: See :func:`~imgaug.augmenters.artistic.stylize_cartoon`. Parameters ---------- blur_ksize : number or tuple of number or list of number or imgaug.parameters.StochasticParameter, optional Median filter kernel size. See :func:`~imgaug.augmenters.artistic.stylize_cartoon` for details. * If ``number``: That value will be used for all images. * If ``tuple (a, b) of number``: A random value will be uniformly sampled per image from the interval ``[a, b)``. * If ``list``: A random value will be picked per image from the ``list``. * If ``StochasticParameter``: The parameter will be queried once per batch for ``(N,)`` values, where ``N`` is the number of images. segmentation_size : number or tuple of number or list of number or imgaug.parameters.StochasticParameter, optional Mean-Shift segmentation size multiplier. See :func:`~imgaug.augmenters.artistic.stylize_cartoon` for details. * If ``number``: That value will be used for all images. * If ``tuple (a, b) of number``: A random value will be uniformly sampled per image from the interval ``[a, b)``. * If ``list``: A random value will be picked per image from the ``list``. * If ``StochasticParameter``: The parameter will be queried once per batch for ``(N,)`` values, where ``N`` is the number of images. saturation : number or tuple of number or list of number or imgaug.parameters.StochasticParameter, optional Saturation multiplier. See :func:`~imgaug.augmenters.artistic.stylize_cartoon` for details. * If ``number``: That value will be used for all images. * If ``tuple (a, b) of number``: A random value will be uniformly sampled per image from the interval ``[a, b)``. * If ``list``: A random value will be picked per image from the ``list``. * If ``StochasticParameter``: The parameter will be queried once per batch for ``(N,)`` values, where ``N`` is the number of images. edge_prevalence : number or tuple of number or list of number or imgaug.parameters.StochasticParameter, optional Multiplier for the prevalence of edges. See :func:`~imgaug.augmenters.artistic.stylize_cartoon` for details. * If ``number``: That value will be used for all images. * If ``tuple (a, b) of number``: A random value will be uniformly sampled per image from the interval ``[a, b)``. * If ``list``: A random value will be picked per image from the ``list``. * If ``StochasticParameter``: The parameter will be queried once per batch for ``(N,)`` values, where ``N`` is the number of images. from_colorspace : str, optional The source colorspace. Use one of ``imgaug.augmenters.color.CSPACE_*``. Defaults to ``RGB``. seed : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. name : None or str, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. random_state : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional Old name for parameter `seed`. Its usage will not yet cause a deprecation warning, but it is still recommended to use `seed` now. Outdated since 0.4.0. deterministic : bool, optional Deprecated since 0.4.0. See method ``to_deterministic()`` for an alternative and for details about what the "deterministic mode" actually does. Examples -------- >>> import imgaug.augmenters as iaa >>> aug = iaa.Cartoon() Create an example image, then apply a cartoon filter to it. >>> aug = iaa.Cartoon(blur_ksize=3, segmentation_size=1.0, >>> saturation=2.0, edge_prevalence=1.0) Create a non-stochastic cartoon augmenter that produces decent-looking images. )rr)g?g333333?)g?g@)g?g?N deprecatedc sztt|j|||| dtj|ddddd|_tj|ddddd|_tj|ddddd|_tj|d dddd|_||_ dS) N)seedname random_state deterministicr0)rNT)Z value_rangeZtuple_to_uniformZlist_to_choicer1)rNr2r3) superrT__init__iapZhandle_continuous_paramr0r1r2r3r4) selfr0r1r2r3r4rVrWrXrY __class__r6r7r[hs*  zCartoon.__init__c Csh|jdur2|||}t|jD]!\}}t||d||d||d||d||jd|d<q|S)Nrrr r r0r1r2r3r4.)Zimages _draw_samples enumerater8r4)r]batchrXparentshooksZsamplesir/r6r6r7_augment_batch_s       zCartoon._augment_batch_cCsJ|j}|jj|f|d|jj|f|d|jj|f|d|jj|f|dfS)N)rX)nb_rowsr0Z draw_samplesr1r2r3)r]rcrXrhr6r6r7ras  zCartoon._draw_samplescCs|j|j|j|j|jgS)z=See :func:`~imgaug.augmenters.meta.Augmenter.get_parameters`.r`)r]r6r6r7get_parameterss zCartoon.get_parameters) __name__ __module__ __qualname____doc__r= CSPACE_RGBr[rgrari __classcell__r6r6r^r7rTsk rT)rm __future__rrrnumpyr%r(Z imgaug.imgaugrrrr=r r r r\rnr8r)r*r'rKr,r-r.Z AugmenterrTr6r6r6r7s.