o e}V@szdZddlmZmZmZddlZddlmZ ddl m Z ddl mZddlmZe jd d d d d ZGdd d eZdS)z^Classes dealing with segmentation maps. E.g. masks, semantic or instance segmentation maps. )print_functiondivisionabsolute_importN)imgaug)blend) IAugmentableSegmentationMapsOnImagez.(Note the plural 'Maps' instead of old 'Map'.)alt_funccommentcOst|i|S)z@Object representing a segmentation map associated with an image.)r )argskwargsrKD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/segmaps.pySegmentationMapOnImagesrc@seZdZdZgdZd%ddZddZejdd d d Z d&d d Z   d'ddZ d(ddZ  d)ddZ ejdddddZd*dd Zd&d!d"Zd&d#d$ZdS)+r a Object representing a segmentation map associated with an image. Attributes ---------- DEFAULT_SEGMENT_COLORS : list of tuple of int Standard RGB colors to use during drawing, ordered by class index. Parameters ---------- arr : (H,W) ndarray or (H,W,C) ndarray Array representing the segmentation map(s). May have dtypes bool, int or uint. shape : tuple of int Shape of the image on which the segmentation map(s) is/are placed. **Not** the shape of the segmentation map(s) array, unless it is identical to the image shape (note the likely difference between the arrays in the number of channels). This is expected to be ``(H, W)`` or ``(H, W, C)`` with ``C`` usually being ``3``. If there is no corresponding image, use ``(H_arr, W_arr)`` instead, where ``H_arr`` is the height of the segmentation map(s) array (analogous ``W_arr``). nb_classes : None or int, optional Deprecated. )*)rrr)K)<r)r)r)r0)r)Fr!)r!2r)rr)r%)rr&)rr%r)n()rr$r)r&rr)r'r)r&r&r)rr)rrr&)r&r&r&)rrr)s %)rZr.)pr-)rAd)zr2)Hr/)#xr9)r9rr,)ir4r)}_r<)r@r=)r,r<r0)U7)r0r;r3)r=rr)r>r0a)r=r=r)r0kr/)rrr=)r=r=r=NcCst|sJdt|f|jdvsJd|j|jft|ts*Jdt||jjdkrJt d|jdkr?|dk}n t j |dd  t j }|jjd krd|j|jf|_|jdkrc|d t jf}nd|jjd vrt |jd dd ksJdt |f|jjdkr|jjdksJd|jjfn|jjdkr|jjdksJd|jjf|j|jf|_|jdkr|d t jf}n td|jjf|jjdkr| t j }||_||_|durt ddSdS)Nz$Expected to get numpy array, got %s.)rzZExpected segmentation map array to be 2- or 3-dimensional, got %d dimensions and shape %s.z|Expected 'shape' to be a tuple denoting the shape of the image on which the segmentation map is placed. Got type %s instead.fzGot a float array as the segmentation map in SegmentationMapsOnImage. That is deprecated. Please provide instead a (H,W,[C]) array of dtype bool_, int or uint, where C denotes the segmentation map index.rg?)Zaxisbool.)iurr3zPExpected segmentation map array to only contain values >=0, got a minimum of %d.rGz]When using uint arrays as segmentation maps, only uint8 and uint16 are allowed. Got dtype %s.rFzaWhen using int arrays as segmentation maps, only int8, int16 and int32 are allowed. Got dtype %s.zSInput was expected to be an array of dtype 'bool', 'int' or 'uint'. Got dtype '%s'.int32zProviding nb_classes to SegmentationMapsOnImage is no longer necessary and hence deprecated. The argument is ignored and can be safely removed.)iaZ is_np_arraytypendimshape isinstancetupledtypekindwarn_deprecatednpZargmaxastyperIname _input_wasZnewaxisminZflatitemsize Exceptionarr)selfrZrM nb_classesrrr__init__fs|                z SegmentationMapsOnImage.__init__cCsV|j\}}|j|}|dkr)|jddksJd|jf|dddddfS|S)aReturn the seg.map array, with original dtype and shape ndim. Here, "original" denotes the dtype and number of shape dimensions that was used when the :class:`SegmentationMapsOnImage` instance was created, i.e. upon the call of :func:`SegmentationMapsOnImage.__init__`. Internally, this class may use a different dtype and shape to simplify computations. .. note:: The height and width may have changed compared to the original input due to e.g. pooling operations. Returns ------- ndarray Segmentation map array. Same dtype and number of dimensions as was originally used when the :class:`SegmentationMapsOnImage` instance was created. rrzOriginally got a (H,W) segmentation map. Internal array should now have shape (H,W,1), but got %s. This might be an internal error.Nr)rVrZrTrM)r[Z input_dtypeZ input_ndimZ arr_inputrrrget_arrs  zSegmentationMapsOnImage.get_arrz!SegmentationMapsOnImage.get_arr())r cOs|S)z=Return the seg.map array, with original dtype and shape ndim.)r^r[rrrrr get_arr_intsz#SegmentationMapsOnImage.get_arr_intcCsdd}|dur ||g}n t|s||g}||d|jjd}||d|jjd}tj||dftjd}|j|dd |d d S) a& Render the segmentation map as an RGB image. Parameters ---------- size : None or float or iterable of int or iterable of float, optional Size of the rendered RGB image as ``(height, width)``. See :func:`~imgaug.imgaug.imresize_single_image` for details. If set to ``None``, no resizing is performed and the size of the segmentation map array is used. colors : None or list of tuple of int, optional Colors to use. One for each class to draw. If ``None``, then default colors will be used. Returns ------- list of (H,W,3) ndarray Rendered segmentation map (dtype is ``uint8``). One per ``C`` in the original input array ``(H,W,C)``. cSsH|dur|St|rtt||dSt|r|Stdt|f)NrzExpected float or int, got %s.)rJZis_single_floatmaxintZis_single_integer ValueErrorrK)ZsizevalZ arr_axis_sizerrr_handle_sizevals  z5SegmentationMapsOnImage.draw.._handle_sizevalNrrrCrPg?segmentation_mapT)alpharesizecolorsdraw_background)rJ is_iterablerZrMrSzerosuint8 draw_on_image)r[sizerirdheightwidthimagerrrdraws  zSegmentationMapsOnImage.draw?rfFrcCs|dur td|jdksJd|jf|jddks'Jd|jdf|jjdks6Jd|jjfd |kr@d ksHnJd |f|d vsSJd |f|durY|ntj}|dkrntj||j jdddd}g}t |j |j jd} | D]} | dddddf} dt | } t j | jd| jddft jd} | t|ksJd| t|ft | } tt| |D]\}}|| vr| |k}|| |<qtj| |jdddd} t| ||}|r|}ntj| |k|jdddd}t |}||||}||q}|S)aDraw the segmentation map as an overlay over an image. Parameters ---------- image : (H,W,3) ndarray Image onto which to draw the segmentation map. Expected dtype is ``uint8``. alpha : float, optional Alpha/opacity value to use for the mixing of image and segmentation map. Larger values mean that the segmentation map will be more visible and the image less visible. resize : {'segmentation_map', 'image'}, optional In case of size differences between the image and segmentation map, either the image or the segmentation map can be resized. This parameter controls which of the two will be resized to the other's size. colors : None or list of tuple of int, optional Colors to use. One for each class to draw. If ``None``, then default colors will be used. draw_background : bool, optional If ``True``, the background will be drawn like any other class. If ``False``, the background will not be drawn, i.e. the respective background pixels will be identical with the image's RGB color at the corresponding spatial location and no color overlay will be applied. background_class_id : int, optional Class id to interpret as the background class. See `draw_background`. background_threshold : None, optional Deprecated. This parameter is ignored. Returns ------- list of (H,W,3) ndarray Rendered overlays as ``uint8`` arrays. Always a **list** containing one RGB image per segmentation map array channel. Nz[The argument `background_threshold` is deprecated and ignored. Please don't use it anymore.rCzFExpected to draw on 3-dimensional image, got image with %d dimensions.rzBExpected to draw on RGB image, got image with %d channels instead.rmz5Expected to get image with dtype uint8, got dtype %s.g:0yEg1?z8Expected 'alpha' to be in interval [0.0, 1.0], got %.4f.)rfrrz>Expected 'resize' to be "segmentation_map" or "image", got %s.rrrZcubic interpolationrrezWCan't draw all %d classes as it would exceed the maximum number of %d available colors.nearest)rJrRrLrMrPrUr DEFAULT_SEGMENT_COLORSimresize_single_imagerZrSZdsplitrarlrmlenuniquezipsmxrangeblendlibZ blend_alphaZ atleast_3dappend)r[rrrgrhrirjZbackground_class_idZbackground_thresholdZ segmaps_drawnZarr_channelwiserZr\Z segmap_drawnZ ids_in_mapccolorZ class_maskZsegmap_on_imageZmixZforeground_maskrrrrns1        z%SegmentationMapsOnImage.draw_on_imageconstantc Cs2ddlm}|j|j||||||d}|j|dS)aPad the segmentation maps at their top/right/bottom/left side. Parameters ---------- top : int, optional Amount of pixels to add at the top side of the segmentation map. Must be ``0`` or greater. right : int, optional Amount of pixels to add at the right side of the segmentation map. Must be ``0`` or greater. bottom : int, optional Amount of pixels to add at the bottom side of the segmentation map. Must be ``0`` or greater. left : int, optional Amount of pixels to add at the left side of the segmentation map. Must be ``0`` or greater. mode : str, optional Padding mode to use. See :func:`~imgaug.imgaug.pad` for details. cval : number, optional Value to use for padding if `mode` is ``constant``. See :func:`~imgaug.imgaug.pad` for details. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Padded segmentation map with height ``H'=H+top+bottom`` and width ``W'=W+left+right``. rro)toprightbottomleftmodecvalrZ) augmentersropadrZdeepcopy) r[rrrrrriasize arr_paddedrrrrs # zSegmentationMapsOnImage.padc CsBddlm}|j|j|||dd\}}|j|d}|r||fS|S)aPad the segmentation maps until they match a target aspect ratio. Depending on which dimension is smaller (height or width), only the corresponding sides (left/right or top/bottom) will be padded. In each case, both of the sides will be padded equally. Parameters ---------- aspect_ratio : float Target aspect ratio, given as width/height. E.g. ``2.0`` denotes the image having twice as much width as height. mode : str, optional Padding mode to use. See :func:`~imgaug.imgaug.pad` for details. cval : number, optional Value to use for padding if `mode` is ``constant``. See :func:`~imgaug.imgaug.pad` for details. return_pad_amounts : bool, optional If ``False``, then only the padded instance will be returned. If ``True``, a tuple with two entries will be returned, where the first entry is the padded instance and the second entry are the amounts by which each array side was padded. These amounts are again a tuple of the form ``(top, right, bottom, left)``, with each value being an integer. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Padded segmentation map as :class:`SegmentationMapsOnImage` instance. tuple of int Amounts by which the instance's array was padded on each side, given as a tuple ``(top, right, bottom, left)``. This tuple is only returned if `return_pad_amounts` was set to ``True``. rrT) aspect_ratiorrreturn_pad_amountsr)rropad_to_aspect_ratiorZr) r[rrrrrrZ pad_amountssegmaprrrrs + z+SegmentationMapsOnImage.pad_to_aspect_ratioz SegmentationMapsOnImage.resize()z(resize() has the exactly same interface.r cOs|j|i|S)zBResize the seg.map(s) array given a target size and interpolation.)rhr_rrrscaleszSegmentationMapsOnImage.scalerwcCstj|j||d}||S)aResize the seg.map(s) array given a target size and interpolation. Parameters ---------- sizes : float or iterable of int or iterable of float New size of the array in ``(height, width)``. See :func:`~imgaug.imgaug.imresize_single_image` for details. interpolation : None or str or int, optional The interpolation to use during resize. Nearest neighbour interpolation (``"nearest"``) is almost always the best choice. See :func:`~imgaug.imgaug.imresize_single_image` for details. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Resized segmentation map object. ru)rJryrZr)r[sizesrvZ arr_resizedrrrrhs  zSegmentationMapsOnImage.resizecCs4t|dur|jn||dur|jn|d}|j|_|S)a~Create a shallow copy of the segmentation map object. Parameters ---------- arr : None or (H,W) ndarray or (H,W,C) ndarray, optional Optionally the `arr` attribute to use for the new segmentation map instance. Will be copied from the old instance if not provided. See :func:`~imgaug.augmentables.segmaps.SegmentationMapsOnImage.__init__` for details. shape : None or tuple of int, optional Optionally the shape attribute to use for the the new segmentation map instance. Will be copied from the old instance if not provided. See :func:`~imgaug.augmentables.segmaps.SegmentationMapsOnImage.__init__` for details. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Shallow copy. NrM)r rZrMrVr[rZrMrrrrcopys zSegmentationMapsOnImage.copycCs:tt|dur |jn||dur|jn|d}|j|_|S)axCreate a deep copy of the segmentation map object. Parameters ---------- arr : None or (H,W) ndarray or (H,W,C) ndarray, optional Optionally the `arr` attribute to use for the new segmentation map instance. Will be copied from the old instance if not provided. See :func:`~imgaug.augmentables.segmaps.SegmentationMapsOnImage.__init__` for details. shape : None or tuple of int, optional Optionally the shape attribute to use for the the new segmentation map instance. Will be copied from the old instance if not provided. See :func:`~imgaug.augmentables.segmaps.SegmentationMapsOnImage.__init__` for details. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Deep copy. Nr)r rSrrZrMrVrrrrrs z SegmentationMapsOnImage.deepcopy)N)NN)rtrfNFrN)rrrrrr)rrF)rw)__name__ __module__ __qualname____doc__rxr]r^rJ deprecatedr`rsrnrrrrhrrrrrrr s. .F #  2 x( 7    )r __future__rrrnumpyrSZ six.movesmovesr}rrJrrrbaser rrr rrrrs