o e)@sdZddlmZmZmZddlZddlZddlZ ddl Z ddl m Z ddlmZddlmZmZmZmZGd d d eZGd d d eZGd ddeZdS)z$Classes representing bounding boxes.)print_functiondivisionabsolute_importN)imgaug) IAugmentable)normalize_shapeproject_coords_remove_out_of_image_fraction__normalize_shift_argsc@seZdZdZdgddZeddZeddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZeddZddZddZddZdhd d!Zdhd"d#Zdgd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zdid4d5Zejd6d7d8d9d:Zd;d<Z d=d>Z!djd?d@Z"dkdAdBZ# C E G 3dldHdIZ$ D 3 dmdJdKZ% E dmdLdMZ&  2dndNdOZ'dPdQZ(dRdSZ)dodUdVZ*dodWdXZ+e,dYdZZ-dpd[d\Z.dpd]d^Z/d_d`Z0dadbZ1dcddZ2dedfZ3dS)q BoundingBoxaClass representing bounding boxes. Each bounding box is parameterized by its top left and bottom right corners. Both are given as x and y-coordinates. The corners are intended to lie inside the bounding box area. As a result, a bounding box that lies completely inside the image but has maximum extensions would have coordinates ``(0.0, 0.0)`` and ``(W - epsilon, H - epsilon)``. Note that coordinates are saved internally as floats. Parameters ---------- x1 : number X-coordinate of the top left of the bounding box. y1 : number Y-coordinate of the top left of the bounding box. x2 : number X-coordinate of the bottom right of the bounding box. y2 : number Y-coordinate of the bottom right of the bounding box. label : None or str, optional Label of the bounding box, e.g. a string representing the class. NcCsF||kr ||}}||kr||}}||_||_||_||_||_dS)z"Create a new BoundingBox instance.Nx1y1x2y2labelselfrrrrrrGD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/bbs.py__init__/s   zBoundingBox.__init__cCsDtjdtjd}|j|jf|dddf<|j|jf|dddf<|S)aGet the top-left and bottom-right coordinates as one array. Added in 0.4.0. Returns ------- ndarray A ``(N, 2)`` numpy array with ``N=2`` containing the top-left and bottom-right coordinates. rrdtyperNr)npemptyfloat32rrrr)rarrrrrcoords<s zBoundingBox.coordscCtt|jS)zGet the x-coordinate of the top left corner as an integer. Returns ------- int X-coordinate of the top left corner, rounded to the closest integer. )intrroundrrrrrx1_intN zBoundingBox.x1_intcCr!)zGet the y-coordinate of the top left corner as an integer. Returns ------- int Y-coordinate of the top left corner, rounded to the closest integer. )r"rr#rr$rrry1_int]r&zBoundingBox.y1_intcCr!)zGet the x-coordinate of the bottom left corner as an integer. Returns ------- int X-coordinate of the bottom left corner, rounded to the closest integer. )r"rr#rr$rrrx2_intlr&zBoundingBox.x2_intcCr!)zGet the y-coordinate of the bottom left corner as an integer. Returns ------- int Y-coordinate of the bottom left corner, rounded to the closest integer. )r"rr#rr$rrry2_int{r&zBoundingBox.y2_intcC |j|jS)zEstimate the height of the bounding box. Returns ------- number Height of the bounding box. )rrr$rrrheight zBoundingBox.heightcCr*)zEstimate the width of the bounding box. Returns ------- number Width of the bounding box. )rrr$rrrwidthr,zBoundingBox.widthcC|j|jdS)zEstimate the x-coordinate of the center point of the bounding box. Returns ------- number X-coordinate of the center point of the bounding box. r)rr-r$rrrcenter_x zBoundingBox.center_xcCr.)zEstimate the y-coordinate of the center point of the bounding box. Returns ------- number Y-coordinate of the center point of the bounding box. r)rr+r$rrrcenter_yr0zBoundingBox.center_ycCs |j|jS)zEstimate the area of the bounding box. Returns ------- number Area of the bounding box, i.e. ``height * width``. )r+r-r$rrrarear,zBoundingBox.areacCsZt|tr |\}}n|j|j}}|j|ko|jkno,|j|ko*|jkSS)aaEstimate whether the bounding box contains a given point. Parameters ---------- other : tuple of number or imgaug.augmentables.kps.Keypoint Point to check for. Returns ------- bool ``True`` if the point is contained in the bounding box, ``False`` otherwise. ) isinstancetuplexyrrrr)rotherr5r6rrrcontainss  8zBoundingBox.containscCs8t|j|jf|j|jfg||\\|_|_\|_|_|S)aProject the bounding box onto a differently shaped image in-place. E.g. if the bounding box is on its original image at ``x1=(10 of 100 pixels)`` and ``y1=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(x1=20, y1=40)``. (Analogous for ``x2``/``y2``.) This is intended for cases where the original image is resized. It cannot be used for more complex changes (e.g. padding, cropping). Added in 0.4.0. Parameters ---------- from_shape : tuple of int or ndarray Shape of the original image. (Before resize.) to_shape : tuple of int or ndarray Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.bbs.BoundingBox ``BoundingBox`` instance with new coordinates. The object may have been modified in-place. )r rrrrrZ from_shapeZto_shaperrrproject_s zBoundingBox.project_cCs|||S)azProject the bounding box onto a differently shaped image. E.g. if the bounding box is on its original image at ``x1=(10 of 100 pixels)`` and ``y1=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(x1=20, y1=40)``. (Analogous for ``x2``/``y2``.) This is intended for cases where the original image is resized. It cannot be used for more complex changes (e.g. padding, cropping). Parameters ---------- from_shape : tuple of int or ndarray Shape of the original image. (Before resize.) to_shape : tuple of int or ndarray Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.bbs.BoundingBox ``BoundingBox`` instance with new coordinates. )deepcopyr:r9rrrprojectszBoundingBox.projectrcCsD|j|||_|j|||_|j|||_|j|||_|S)aExtend the size of the bounding box along its sides in-place. Added in 0.4.0. Parameters ---------- all_sides : number, optional Value by which to extend the bounding box size along all sides. top : number, optional Value by which to extend the bounding box size along its top side. right : number, optional Value by which to extend the bounding box size along its right side. bottom : number, optional Value by which to extend the bounding box size along its bottom side. left : number, optional Value by which to extend the bounding box size along its left side. Returns ------- imgaug.BoundingBox Extended bounding box. The object may have been modified in-place. rrrrrZ all_sidestoprightbottomleftrrrextend_s "zBoundingBox.extend_cCs||||||S)a9Extend the size of the bounding box along its sides. Parameters ---------- all_sides : number, optional Value by which to extend the bounding box size along all sides. top : number, optional Value by which to extend the bounding box size along its top side. right : number, optional Value by which to extend the bounding box size along its right side. bottom : number, optional Value by which to extend the bounding box size along its bottom side. left : number, optional Value by which to extend the bounding box size along its left side. Returns ------- imgaug.BoundingBox Extended bounding box. )r;rCr>rrrextendDszBoundingBox.extendcCs\t|j|j}t|j|j}t|j|j}t|j|j}||ks$||kr&|St||||dS)aLCompute the intersection BB between this BB and another BB. Note that in extreme cases, the intersection can be a single point. In that case the intersection bounding box exists and it will be returned, but it will have a height and width of zero. Parameters ---------- other : imgaug.augmentables.bbs.BoundingBox Other bounding box with which to generate the intersection. default : any, optional Default value to return if there is no intersection. Returns ------- imgaug.augmentables.bbs.BoundingBox or any Intersection bounding box of the two bounding boxes if there is an intersection. If there is no intersection, the default value will be returned, which can by anything. rrrr)maxrrminrrr )rr7defaultZx1_iZy1_iZx2_iZy2_irrr intersectioneszBoundingBox.intersectioncCs8tt|j|jt|j|jt|j|jt|j|jdS)aCompute the union BB between this BB and another BB. This is equivalent to drawing a bounding box around all corner points of both bounding boxes. Parameters ---------- other : imgaug.augmentables.bbs.BoundingBox Other bounding box with which to generate the union. Returns ------- imgaug.augmentables.bbs.BoundingBox Union bounding box of the two bounding boxes. rE)r rGrrrFrr)rr7rrrunions     zBoundingBox.unioncCs>||}|dur dS|j|j|j}|dkr|j|SdS)aCompute the IoU between this bounding box and another one. IoU is the intersection over union, defined as:: ``area(intersection(A, B)) / area(union(A, B))`` ``= area(intersection(A, B)) / (area(A) + area(B) - area(intersection(A, B)))`` Parameters ---------- other : imgaug.augmentables.bbs.BoundingBox Other bounding box with which to compare. Returns ------- float IoU between the two bounding boxes. Nr)rIr2)rr7ZintersZ area_unionrrrious zBoundingBox.ioucCsRt|}|dd\}}tdd||d}|j|dd}|j}|dur$|S||jS)a+Compute the area of the BB that is outside of the image plane. Added in 0.4.0. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- float Total area of the bounding box that is outside of the image plane. Can be ``0.0``. rrrEN)rH)r r rIr2)rimageshaper+r-Zbb_imageinterr2rrrcompute_out_of_image_areas z%BoundingBox.compute_out_of_image_areac Csp|j}|dkr1t|}|dd\}}|jdkp|j|k}|jdkp&|j|k}|p*|}|r/dSdS|||S)aCompute fraction of BB area outside of the image plane. This estimates ``f = A_ooi / A``, where ``A_ooi`` is the area of the bounding box that is outside of the image plane, while ``A`` is the total area of the bounding box. Added in 0.4.0. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- float Fraction of the bounding box area that is outside of the image plane. Returns ``0.0`` if the bounding box is fully inside of the image plane. If the bounding box has an area of zero, the result is ``1.0`` if its coordinates are outside of the image plane, otherwise ``0.0``. rr?rK)r2r rrrP) rrMr2rNr+r-Z y1_outsideZ x1_outsideZ is_outsiderrrcompute_out_of_image_fractions z)BoundingBox.compute_out_of_image_fractioncCs@t|}|dd\}}|jdko|j|ko|jdko|j|kS)aEstimate whether the bounding box is fully inside the image area. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- bool ``True`` if the bounding box is fully inside the image area. ``False`` otherwise. rr)r rrrr)rrMrNr+r-rrris_fully_within_images z!BoundingBox.is_fully_within_imagecCsLt|}|dd\}}ttjj}td||d||d}||duS)a/Estimate whether the BB is at least partially inside the image area. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- bool ``True`` if the bounding box is at least partially inside the image area. ``False`` otherwise. rrr=N)r rfinforepsr rI)rrMrNr+r-rUZimg_bbrrris_partly_within_images z"BoundingBox.is_partly_within_imageTFcCs ||rdS||r|S|S)avEstimate whether the BB is partially/fully outside of the image area. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. fully : bool, optional Whether to return ``True`` if the bounding box is fully outside of the image area. partly : bool, optional Whether to return ``True`` if the bounding box is at least partially outside fo the image area. Returns ------- bool ``True`` if the bounding box is partially/fully outside of the image area, depending on defined parameters. ``False`` otherwise. F)rSrV)rrMfullypartlyrrris_out_of_image*s  zBoundingBox.is_out_of_imagezBoundingBox.clip_out_of_image()3clip_out_of_image() has the exactly same interface.Zalt_funccommentcOs|j|i|S)z?Clip off all parts of the BB box that are outside of the image.clip_out_of_image)rargskwargsrrrcut_out_of_imageKszBoundingBox.cut_out_of_imagecCst|}|dd\}}|dksJd|jf|dks$Jd|jfttjj}t|jd|||_t|jd|||_t|j d|||_ t|j d|||_ |S)aClip off parts of the BB box that are outside of the image in-place. Added in 0.4.0. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use for the clipping of the bounding box. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- imgaug.augmentables.bbs.BoundingBox Bounding box, clipped to fall within the image dimensions. The object may have been modified in-place. rrz+Expected image with height>0, got shape %s.z*Expected image with width>0, got shape %s.) r rNrrTrrUcliprrrr)rrMrNr+r-rUrrrclip_out_of_image_Rs    zBoundingBox.clip_out_of_image_cC||S)a2Clip off all parts of the BB box that are outside of the image. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use for the clipping of the bounding box. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- imgaug.augmentables.bbs.BoundingBox Bounding box, clipped to fall within the image dimensions. r;rcrrMrrrr^vzBoundingBox.clip_out_of_imagecCs<|j|7_|j|7_|j|7_|j|7_|S)aMove this bounding box along the x/y-axis in-place. The origin ``(0, 0)`` is at the top left of the image. Added in 0.4.0. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. Returns ------- imgaug.augmentables.bbs.BoundingBox Shifted bounding box. The object may have been modified in-place. r=)rr5r6rrrshift_s zBoundingBox.shift_cCs(t||||||d\}}|||S)aMove this bounding box along the x/y-axis. The origin ``(0, 0)`` is at the top left of the image. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. top : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the top (towards the bottom). right : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the right (towards the left). bottom : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the bottom (towards the top). left : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the left (towards the right). Returns ------- imgaug.augmentables.bbs.BoundingBox Shifted bounding box. r?r@rArBr r;rhrr5r6r?r@rArBrrrshifts*  zBoundingBox.shiftrrrQrc Cs4t|||||| ||d} | r| ||S| ||S)a,Draw a box showing the BB's label. The box is placed right above the BB's rectangle. Added in 0.4.0. Parameters ---------- image : (H,W,C) ndarray The image onto which to draw the label. Currently expected to be ``uint8``. color : None or iterable of int, optional The color to use, corresponding to the channel layout of the image. Usually RGB. Text and background colors will be derived from this. color_text : None or iterable of int, optional The text color to use. If ``None``, derived from `color_bg`. color_bg : None or iterable of int, optional The background color of the label box. If ``None``, derived from `color`. alpha : float, optional The transparency of the drawn bounding box, where ``1.0`` denotes no transparency and ``0.0`` is invisible. size : int, optional The thickness of the bounding box in pixels. If the value is larger than ``1``, then additional pixels will be added around the bounding box (i.e. extension towards the outside). size_text : int, optional Font size to use. height : int, optional Height of the label box in pixels. copy : bool, optional Whether to copy the input image or change it in-place. raise_if_out_of_image : bool, optional Whether to raise an error if the bounding box is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- (H,W,C) ndarray(uint8) Image with bounding box drawn on it. color color_textcolor_bgsizealpharaise_if_out_of_imager+ size_text)_LabelOnImageDrawer draw_on_imagedraw_on_image_) rrMrrrsrtrvrurxr+copyrwZdrawerrrrdraw_label_on_images;  zBoundingBox.draw_label_on_imagecCs|dur td|}|r"||r"td|j|j|j|j|jf|r)t |n|}t |t t fr7t |}t|D]} |j|j|j|jf\} } } } ||rt | d|jdd} t | d|jdd} t | d|jdd} t | d|jdd} | | | | | | | | g}| | | | | | | | g}tjj|||jd\}}|dkr||||ddf<q;t|rd||||ddf|||||ddf<t |dd}q;|j}|t j}d||||ddf|||||ddf<t |dd|}q;|S) aDraw the rectangle of the bounding box on an image. This method does not draw the label. Added in 0.4.0. Parameters ---------- image : (H,W,C) ndarray The image onto which to draw the bounding box rectangle. Currently expected to be ``uint8``. color : iterable of int, optional The color to use, corresponding to the channel layout of the image. Usually RGB. alpha : float, optional The transparency of the drawn bounding box, where ``1.0`` denotes no transparency and ``0.0`` is invisible. size : int, optional The thickness of the bounding box in pixels. If the value is larger than ``1``, then additional pixels will be added around the bounding box (i.e. extension towards the outside). copy : bool, optional Whether to copy the input image or change it in-place. raise_if_out_of_image : bool, optional Whether to raise an error if the bounding box is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. thickness : None or int, optional Deprecated. Returns ------- (H,W,C) ndarray(uint8) Image with bounding box drawn on it. NzoUsage of argument 'thickness' in BoundingBox.draw_on_image() is deprecated. The argument was renamed to 'size'.SCannot draw bounding box x1=%.8f, y1=%.8f, x2=%.8f, y2=%.8f on image with shape %s.rrrNGz?rn)iaZwarn_deprecatedrY ExceptionrrrrrNrr|r3r4listuint8ranger'r)r%r(rSrbskimageZdrawZpolygon_perimeterZis_float_arrayrastyper)rrMrrrvrur|rw thicknessresultirrrrr6r5rrcc input_dtyperrrdraw_box_on_imagesR.     zBoundingBox.draw_box_on_imagec CsH|j|||||||d}|jdur"|j||||dur|n|d|d}|S)aDraw the bounding box on an image. This will automatically also draw the label, unless it is ``None``. To only draw the box rectangle use :func:`~imgaug.augmentables.bbs.BoundingBox.draw_box_on_image`. To draw the label even if it is ``None`` or to configure e.g. its color, use :func:`~imgaug.augmentables.bbs.BoundingBox.draw_label_on_image`. Parameters ---------- image : (H,W,C) ndarray The image onto which to draw the bounding box. Currently expected to be ``uint8``. color : iterable of int, optional The color to use, corresponding to the channel layout of the image. Usually RGB. alpha : float, optional The transparency of the drawn bounding box, where ``1.0`` denotes no transparency and ``0.0`` is invisible. size : int, optional The thickness of the bounding box in pixels. If the value is larger than ``1``, then additional pixels will be added around the bounding box (i.e. extension towards the outside). copy : bool, optional Whether to copy the input image or change it in-place. raise_if_out_of_image : bool, optional Whether to raise an error if the bounding box is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. thickness : None or int, optional Deprecated. Returns ------- (H,W,C) ndarray(uint8) Image with bounding box drawn on it. rrrvrur|rwrNF)rrrvrur|rw)rrr}) rrMrrrvrur|rwrZ image_drawnrrrrzs0 zBoundingBox.draw_on_imagecCsf|jd|jd}}|j|j|j|jf\}}} } ||} | r:t| | gd|d\} } t||gd|d\}}|rTt||dkrH|d}t| | dkrT| d} |rddl m } d} d}d}d}|dkrvt|}||}||}d}| dkrt| } | | } || }d} ||kr||}| |kr| |}| |||g}t dd|D}|r|durt |}| j |t| |t||t||t||d}|| | ||fSd || || fko||||fkn}| | ||}}|dk}|dk}|r |r |r || | ||fS|rd}d}nd}d}|jdkr%tj||f|jd Stj|||jd f|jd S) aExtract the image pixels within the bounding box. This function will zero-pad the image if the bounding box is partially/fully outside of the image. Parameters ---------- image : (H,W) ndarray or (H,W,C) ndarray The image from which to extract the pixels within the bounding box. pad : bool, optional Whether to zero-pad the image if the object is partially/fully outside of it. pad_max : None or int, optional The maximum number of pixels that may be zero-paded on any side, i.e. if this has value ``N`` the total maximum of added pixels is ``4*N``. This option exists to prevent extremely large images as a result of single points being moved very far away during augmentation. prevent_zero_size : bool, optional Whether to prevent the height or width of the extracted image from becoming zero. If this is set to ``True`` and the height or width of the bounding box is below ``1``, the height/width will be increased to ``1``. This can be useful to prevent problems, e.g. with image saving or plotting. If it is set to ``False``, images will be returned as ``(H', W')`` or ``(H', W', 3)`` with ``H`` or ``W`` potentially being 0. Returns ------- (H',W') ndarray or (H',W',C) ndarray Pixels within the bounding box. Zero-padded if the bounding box is partially/fully outside of the image. If `prevent_zero_size` is activated, it is guarantueed that ``H'>0`` and ``W'>0``, otherwise only ``H'>=0`` and ``W'>=0``. rrr)rucSsg|]}|dkqS)rr).0valrrr z2BoundingBox.extract_from_image..Nri)rrrrr)rNr%r(r'r)rSrrbabsZ augmentersruanyrFpadrGndimzerosr)rrMrZpad_maxZprevent_zero_sizer+r-rrrrZ fully_withinZiasizepad_top pad_rightZ pad_bottompad_leftZpaddingsZ any_paddedZ within_imageZ out_heightZ out_widthZnonzero_heightZ nonzero_widthrrrextract_from_images+      zBoundingBox.extract_from_imagecCsHddlm}||j|jd||j|jd||j|jd||j|jdgS)zConvert the BB's corners to keypoints (clockwise, from top left). Returns ------- list of imgaug.augmentables.kps.Keypoint Corners of the bounding box as keypoints. r)Keypointr5r6)Zimgaug.augmentables.kpsrrrrr)rrrrr to_keypointsCs zBoundingBox.to_keypointscCsBddlm}||j|jf|j|jf|j|jf|j|jfg|jdS)zConvert this bounding box to a polygon covering the same area. Added in 0.4.0. Returns ------- imgaug.augmentables.polys.Polygon The bounding box converted to a polygon. r)Polygon)r)Zimgaug.augmentables.polysrrrrrr)rrrrr to_polygonVs     zBoundingBox.to_polygon-C6?cCslt|tr |jj}nt|r|j}nt|r tt|}n t dt |f|j}t j |j||ddS)aEstimate if this and another BB have almost identical coordinates. Added in 0.4.0. Parameters ---------- other : imgaug.augmentables.bbs.BoundingBox or iterable The other bounding box with which to compare this one. If this is an ``iterable``, it is assumed to represent the top-left and bottom-right coordinates of that bounding box, given as e.g. an ``(2,2)`` ndarray or an ``(4,)`` ndarray or as a similar list. max_distance : number, optional The maximum euclidean distance between a corner on one bounding box and the closest corner on the other bounding box. If the distance is exceeded for any such pair, the two BBs are not viewed as equal. Returns ------- bool Whether the two bounding boxes have almost identical corner coordinates. zgExpected 'other' to be an iterable containing two (x,y)-coordinate pairs or a BoundingBox. Got type %s.r)ZatolZrtol) r3r r ZflatrZ is_np_array is_iterablerflatten ValueErrortyperZallclose)rr7 max_distanceZcoords_bZcoords_arrrcoords_almost_equalsls    z BoundingBox.coords_almost_equalscCs|j|jkrdS|j||dS)aCompare this and another BB's label and coordinates. This is the same as :func:`~imgaug.augmentables.bbs.BoundingBox.coords_almost_equals` but additionally compares the labels. Added in 0.4.0. Parameters ---------- other : imgaug.augmentables.bbs.BoundingBox or iterable The other object to compare against. Expected to be a ``BoundingBox``. max_distance : number, optional See :func:`~imgaug.augmentables.bbs.BoundingBox.coords_almost_equals`. Returns ------- bool ``True`` if the coordinates are almost equal and additionally the labels are equal. Otherwise ``False``. F)r)rr)rr7rrrr almost_equalss zBoundingBox.almost_equalscCstj|tjd}t|dksJd|jf|jdks/|jdkr'|jddks/Jd|jf|jdkr9|d}tj|dd \}}tj|dd \}}|||||d S) awConvert a ``(2P,) or (P,2) ndarray`` to a BB instance. This is the inverse of :func:`~imgaug.BoundingBoxesOnImage.to_xyxy_array`. Added in 0.4.0. Parameters ---------- xy : (2P,) ndarray or (P, 2) array or iterable of number or iterable of iterable of number Array containing ``P`` points in xy-form denoting a soup of points around which to place a bounding box. The array should usually be of dtype ``float32``. Returns ------- imgaug.augmentables.bbs.BoundingBox Bounding box around the points. rrzPExpected to get at least one point to place a bounding box around, got shape %s.rrrz;Expected input array of shape (P,) or (P, 2), got shape %s.rr)ZaxisrE) rarrayrlenrNrreshaperGrF)clsxyrrrrrrrfrom_point_soups $  zBoundingBox.from_point_soupcCsbt|dur|jn||dur|jn||dur|jn||dur |jn||dur-t|jdS|dS)aCreate a shallow copy of this BoundingBox instance. Parameters ---------- x1 : None or number If not ``None``, then the ``x1`` coordinate of the copied object will be set to this value. y1 : None or number If not ``None``, then the ``y1`` coordinate of the copied object will be set to this value. x2 : None or number If not ``None``, then the ``x2`` coordinate of the copied object will be set to this value. y2 : None or number If not ``None``, then the ``y2`` coordinate of the copied object will be set to this value. label : None or string If not ``None``, then the ``label`` of the copied object will be set to this value. Returns ------- imgaug.augmentables.bbs.BoundingBox Shallow copy. N)rrrrr)r rrrrr|r;rrrrrr|szBoundingBox.copycCs|j|||||dS)a Create a deep copy of the BoundingBox object. Parameters ---------- x1 : None or number If not ``None``, then the ``x1`` coordinate of the copied object will be set to this value. y1 : None or number If not ``None``, then the ``y1`` coordinate of the copied object will be set to this value. x2 : None or number If not ``None``, then the ``x2`` coordinate of the copied object will be set to this value. y2 : None or number If not ``None``, then the ``y2`` coordinate of the copied object will be set to this value. label : None or string If not ``None``, then the ``label`` of the copied object will be set to this value. Returns ------- imgaug.augmentables.bbs.BoundingBox Deep copy. r)r|rrrrr;s"zBoundingBox.deepcopycC |j|S)zGet the coordinate(s) with given indices. Added in 0.4.0. Returns ------- ndarray xy-coordinate(s) as ``ndarray``. )r rindicesrrr __getitem__) zBoundingBox.__getitem__cC t|jS)zIterate over the coordinates of this instance. Added in 0.4.0. Yields ------ ndarray An ``(2,)`` ``ndarray`` denoting an xy-coordinate pair. )iterr r$rrr__iter__6rzBoundingBox.__iter__cC|SN__str__r$rrr__repr__CzBoundingBox.__repr__cCsd|j|j|j|j|jfS)Nz9BoundingBox(x1=%.4f, y1=%.4f, x2=%.4f, y2=%.4f, label=%s)rr$rrrrFszBoundingBox.__str__r)rrrrrTFrrrrNNNN) rmNNrQrrorpTFrmrQrTFN)TNT)r)NNNNN)4__name__ __module__ __qualname____doc__rpropertyr r%r'r(r)r+r-r/r1r2r8r:r<rCrDrIrJrLrPrRrSrVrYr deprecatedrarcr^rhrlr}rrzrrrrr classmethodrr|r;rrrrrrrrr s            $  ( ! % ! $  . H e >   +  ( '$  r c@sdeZdZdZddZeddZejddZeddZed d Z ed d Z d dZ ddZ e ddZe ddZejfddZddZddZddZ  "dOd#d$ZdPd%d&ZdPd'd(Zd)d*Zd+d,Zejd-d.d/d0d1Zd2d3Zd4d5ZdQd7d8Z dRd9d:Z!d;d<Z"d=d>Z#d?d@Z$dSdAdBZ%dSdCdDZ&dEdFZ'dGdHZ(dIdJZ)dKdLZ*dMdNZ+d"S)TBoundingBoxesOnImageaContainer for the list of all bounding boxes on a single image. Parameters ---------- bounding_boxes : list of imgaug.augmentables.bbs.BoundingBox List of bounding boxes on the image. shape : tuple of int or ndarray The shape of the image on which the objects are placed. Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. Examples -------- >>> import numpy as np >>> from imgaug.augmentables.bbs import BoundingBox, BoundingBoxesOnImage >>> >>> image = np.zeros((100, 100)) >>> bbs = [ >>> BoundingBox(x1=10, y1=20, x2=20, y2=30), >>> BoundingBox(x1=25, y1=50, x2=30, y2=70) >>> ] >>> bbs_oi = BoundingBoxesOnImage(bbs, shape=image.shape) cCs||_t||_dSr)bounding_boxesr rNrrrNrrrreszBoundingBoxesOnImage.__init__cCs|jS)zGet the bounding boxes in this container. Added in 0.4.0. Returns ------- list of BoundingBox Bounding boxes within this container. rr$rrritemsis zBoundingBoxesOnImage.itemscCs ||_dS)zSet the bounding boxes in this container. Added in 0.4.0. Parameters ---------- value : list of BoundingBox Bounding boxes within this container. Nr)rvaluerrrrws cC |jdS)zGet the height of the image on which the bounding boxes fall. Returns ------- int Image height. rrr$rrrr+ zBoundingBoxesOnImage.heightcCr)zGet the width of the image on which the bounding boxes fall. Returns ------- int Image width. rrr$rrrr-rzBoundingBoxesOnImage.widthcCst|jdkS)zDetermine whether this instance contains zero bounding boxes. Returns ------- bool True if this object contains zero bounding boxes. r)rrr$rrrrs zBoundingBoxesOnImage.emptycCs^t|}|dd|jddkr||_|St|jD]\}}||j||j|<q||_|S)a<Project BBs from one image (shape) to a another one in-place. Added in 0.4.0. Parameters ---------- image : ndarray or tuple of int New image onto which the bounding boxes are to be projected. May also simply be that new image's shape tuple. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Object containing the same bounding boxes after projection to the new image shape. The object and its items may have been modified in-place. rr)r rN enumeraterr:r)rrMZon_shaperitemrrron_szBoundingBoxesOnImage.on_cCrd)aProject bounding boxes from one image (shape) to a another one. Parameters ---------- image : ndarray or tuple of int New image onto which the bounding boxes are to be projected. May also simply be that new image's shape tuple. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Object containing the same bounding boxes after projection to the new image shape. )r;rrfrrronrgzBoundingBoxesOnImage.oncCstj|tjd}|jddkrtg|S|jdkr |jddks6|jdkr.|jdddks6Jd |jf|d }d d |D}|||S) afConvert an ``(N, 4) or (N, 2, 2) ndarray`` to a BBsOI instance. This is the inverse of :func:`~imgaug.BoundingBoxesOnImage.to_xyxy_array`. Parameters ---------- xyxy : (N, 4) ndarray or (N, 2, 2) array Array containing the corner coordinates of ``N`` bounding boxes. Each bounding box is represented by its top-left and bottom-right coordinates. The array should usually be of dtype ``float32``. shape : tuple of int Shape of the image on which the bounding boxes are placed. Should usually be ``(H, W, C)`` or ``(H, W)``. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Object containing a list of :class:`BoundingBox` instances derived from the provided corner coordinates. rrrrrrz@Expected input array of shape (N, 4) or (N, 2, 2), got shape %s.)rrrcSg|]}t|qSrr rrrowrrrrz8BoundingBoxesOnImage.from_xyxy_array..)rrrrNrrr)rxyxyrNboxesrrrfrom_xyxy_arrays   z$BoundingBoxesOnImage.from_xyxy_arraycCs(tj|tjd}dd|D}|||S)aSConvert an ``(N, 2P) or (N, P, 2) ndarray`` to a BBsOI instance. Added in 0.4.0. Parameters ---------- xy : (N, 2P) ndarray or (N, P, 2) array or iterable of iterable of number or iterable of iterable of iterable of number Array containing the corner coordinates of ``N`` bounding boxes. Each bounding box is represented by a soup of ``P`` points. If ``(N, P)`` then the second axis is expected to be in xy-form (e.g. ``x1``, ``y1``, ``x2``, ``y2``, ...). The final bounding box coordinates will be derived using ``min`` and ``max`` operations on the xy-values. The array should usually be of dtype ``float32``. shape : tuple of int Shape of the image on which the bounding boxes are placed. Should usually be ``(H, W, C)`` or ``(H, W)``. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Object containing a list of :class:`BoundingBox` instances derived from the provided point soups. rcSrrrrrrrr*rz9BoundingBoxesOnImage.from_point_soups..)rrr)rrrNrrrrfrom_point_soups s z%BoundingBoxesOnImage.from_point_soupscCsPtjt|jdftjd}t|jD]\}}|j|j|j|j g||<q| |S)a4Convert the ``BoundingBoxesOnImage`` object to an ``(N,4) ndarray``. This is the inverse of :func:`~imgaug.BoundingBoxesOnImage.from_xyxy_array`. Parameters ---------- dtype : numpy.dtype, optional Desired output datatype of the ndarray. Returns ------- ndarray ``(N,4) ndarray``, where ``N`` denotes the number of bounding boxes and ``4`` denotes the top-left and bottom-right bounding box corner coordinates in form ``(x1, y1, x2, y2)``. rr) rrrrrrrrrrr)rrZ xyxy_arrayrboxrrr to_xyxy_array.s z"BoundingBoxesOnImage.to_xyxy_arraycCs|dS)a Convert the ``BoundingBoxesOnImage`` object to an ``(N,2) ndarray``. Added in 0.4.0. Returns ------- ndarray ``(2*B,2) ndarray`` of xy-coordinates, where ``B`` denotes the number of bounding boxes. r)rrr$rrr to_xy_arrayHs z BoundingBoxesOnImage.to_xy_arraycCstj|tjd}|jddks#|jdkr|jddks#Jd|jft|t|jks9Jdt|t|jft|j|D]$\}\}}}}t||g|_ t||g|_ t ||g|_ t ||g|_ q?|S)aModify the BB coordinates of this instance in-place. .. note:: This currently expects exactly one entry in `xyxy` per bounding in this instance. (I.e. two corner coordinates per instance.) Otherwise, an ``AssertionError`` will be raised. .. note:: This method will automatically flip x-coordinates if ``x1>x2`` for a bounding box. (Analogous for y-coordinates.) Added in 0.4.0. Parameters ---------- xyxy : (N, 4) ndarray or iterable of iterable of number Coordinates of ``N`` bounding boxes on an image, given as a ``(N,4)`` array of two corner xy-coordinates per bounding box. ``N`` must match the number of bounding boxes in this instance. Returns ------- BoundingBoxesOnImage This instance itself, with updated bounding box coordinates. Note that the instance was modified in-place. rrrrrz7Expected input array to have shape (N,4), got shape %s.zsExpected to receive an array with as many rows there are bounding boxes in this instance. Got %d rows, expected %d.)rrrrNrrrziprGrrrFrr)rrbbrrrrrrrfill_from_xyxy_array_Vs$(z*BoundingBoxesOnImage.fill_from_xyxy_array_cCs tj|tjd}||dS)aModify the BB coordinates of this instance in-place. See :func:`~imgaug.augmentables.bbs.BoundingBoxesOnImage.fill_from_xyxy_array_`. Added in 0.4.0. Parameters ---------- xy : (2*B, 2) ndarray or iterable of iterable of number Coordinates of ``B`` bounding boxes on an image, given as a ``(2*B,2)`` array of two corner xy-coordinates per bounding box. ``B`` must match the number of bounding boxes in this instance. Returns ------- BoundingBoxesOnImage This instance itself, with updated bounding box coordinates. Note that the instance was modified in-place. r)rr)rrrrr)rrrrrfill_from_xy_array_sz(BoundingBoxesOnImage.fill_from_xy_array_rmrQrTFNc Cs:|rt|n|}|jD]}|j||||d||d}q |S)a/Draw all bounding boxes onto a given image. Parameters ---------- image : (H,W,3) ndarray The image onto which to draw the bounding boxes. This image should usually have the same shape as set in ``BoundingBoxesOnImage.shape``. color : int or list of int or tuple of int or (3,) ndarray, optional The RGB color of all bounding boxes. If a single ``int`` ``C``, then that is equivalent to ``(C,C,C)``. alpha : float, optional Alpha/transparency of the bounding box. size : int, optional Thickness in pixels. copy : bool, optional Whether to copy the image before drawing the bounding boxes. raise_if_out_of_image : bool, optional Whether to raise an exception if any bounding box is outside of the image. thickness : None or int, optional Deprecated. Returns ------- (H,W,3) ndarray Image with drawn bounding boxes. Fr)rr|rrz) rrMrrrvrur|rwrrrrrrzs&  z"BoundingBoxesOnImage.draw_on_imagecsfddjD_S)aRemove in-place all BBs that are fully/partially outside of the image. Added in 0.4.0. Parameters ---------- fully : bool, optional Whether to remove bounding boxes that are fully outside of the image. partly : bool, optional Whether to remove bounding boxes that are partially outside of the image. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Reduced set of bounding boxes, with those that were fully/partially outside of the image being removed. The object and its items may have been modified in-place. cs"g|] }|jjds|qS)rWrX)rYrNrrrWrXrrrrs z=BoundingBoxesOnImage.remove_out_of_image_..rrrWrXrrrremove_out_of_image_sz)BoundingBoxesOnImage.remove_out_of_image_cCs|j||dS)aDRemove all BBs that are fully/partially outside of the image. Parameters ---------- fully : bool, optional Whether to remove bounding boxes that are fully outside of the image. partly : bool, optional Whether to remove bounding boxes that are partially outside of the image. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Reduced set of bounding boxes, with those that were fully/partially outside of the image being removed. r)r|rrrrrremove_out_of_imagesz(BoundingBoxesOnImage.remove_out_of_imagecCs t||S)aRemove in-place all BBs with an OOI fraction of at least `fraction`. 'OOI' is the abbreviation for 'out of image'. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a bounding box has to have in order to be removed. A fraction of ``1.0`` removes only bounding boxes that are ``100%`` outside of the image. A fraction of ``0.0`` removes all bounding boxes. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Reduced set of bounding boxes, with those that had an out of image fraction greater or equal the given one removed. The object and its items may have been modified in-place. )r rfractionrrrremove_out_of_image_fraction_ s z2BoundingBoxesOnImage.remove_out_of_image_fraction_cCrd)aRemove all BBs with an out of image fraction of at least `fraction`. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a bounding box has to have in order to be removed. A fraction of ``1.0`` removes only bounding boxes that are ``100%`` outside of the image. A fraction of ``0.0`` removes all bounding boxes. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Reduced set of bounding boxes, with those that had an out of image fraction greater or equal the given one removed. )r|rrrrrremove_out_of_image_fraction#sz1BoundingBoxesOnImage.remove_out_of_image_fractionz(BoundingBoxesOnImage.clip_out_of_image()rZr[cCr)z>Clip off all parts from all BBs that are outside of the image.r]r$rrrra9sz%BoundingBoxesOnImage.cut_out_of_imagecs@fddjD_tjD] \}}|jj|<qS)aX Clip off in-place all parts from all BBs that are outside of the image. Added in 0.4.0. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Bounding boxes, clipped to fall within the image dimensions. The object and its items may have been modified in-place. csg|] }|jr|qSr)rVrNrr$rrrNs  z;BoundingBoxesOnImage.clip_out_of_image_..)rrr^rN)rrrrr$rrc@sz'BoundingBoxesOnImage.clip_out_of_image_cCs |S)zClip off all parts from all BBs that are outside of the image. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Bounding boxes, clipped to fall within the image dimensions. rer$rrrr^Vs z&BoundingBoxesOnImage.clip_out_of_imagercCs,t|jD]\}}|j||d|j|<q|S)aMove all BBs along the x/y-axis in-place. The origin ``(0, 0)`` is at the top left of the image. Added in 0.4.0. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Shifted bounding boxes. The object and its items may have been modified in-place. r)rrrh)rr5r6rrrrrrhaszBoundingBoxesOnImage.shift_cCs*t||||||d\}}|j||dS)aMove all BBs along the x/y-axis. The origin ``(0, 0)`` is at the top left of the image. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. top : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the top (towards the bottom). right : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the right (towads the left). bottom : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the bottom (towards the top). left : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the left (towards the right). Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Shifted bounding boxes. rirrjrkrrrrl}s)  zBoundingBoxesOnImage.shiftc Csxddlm}tjt|jdftjd}t|jD]\}}|j|j |j |j |j |j |j|j g||<q|j | d|jdS)aJConvert the bounding boxes to one ``KeypointsOnImage`` instance. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage A keypoints instance containing ``N*4`` coordinates for ``N`` bounding boxes. Order matches the order in ``bounding_boxes``. r)KeypointsOnImagerrr)ZkpsrrrrrrrrrrrZ from_xy_arrayrrN)rrrrrrrrto_keypoints_on_images  z*BoundingBoxesOnImage.to_keypoints_on_imagecCst|jt|jdksJdt|jdt|jft|jD]d\}}|jd|dj|jd|dj|jd|dj|jd|djg}|jd|dj|jd|dj|jd|dj|jd|djg}t||_t||_t ||_ t ||_ q!|j |_ |S)aJInvert the output of ``to_keypoints_on_image()`` in-place. This function writes in-place into this ``BoundingBoxesOnImage`` instance. Added in 0.4.0. Parameters ---------- kpsoi : imgaug.augmentables.kps.KeypointsOnImages Keypoints to convert back to bounding boxes, i.e. the outputs of ``to_keypoints_on_image()``. Returns ------- BoundingBoxesOnImage Bounding boxes container with updated coordinates. Note that the instance is also updated in-place. rz Expected %d coordinates, got %d.rrrr) rZ keypointsrrr5r6rGrrrFrrrN)rZkpsoirrxxyyrrrinvert_to_keypoints_on_image_s$$$$$    z2BoundingBoxesOnImage.invert_to_keypoints_on_image_cCs*ddlm}dd|jD}|||jdS)a=Convert the bounding boxes to one ``PolygonsOnImage`` instance. Added in 0.4.0. Returns ------- imgaug.augmentables.polys.PolygonsOnImage A ``PolygonsOnImage`` containing polygons. Each polygon covers the same area as the corresponding bounding box. r)PolygonsOnImagecSg|]}|qSr)rrrrrrrz=BoundingBoxesOnImage.to_polygons_on_image..r)ZpolysrrrN)rrZpolygonsrrrto_polygons_on_images z)BoundingBoxesOnImage.to_polygons_on_imagecCs2|dur |jdd}|durt|j}t||S)a_Create a shallow copy of the ``BoundingBoxesOnImage`` instance. Parameters ---------- bounding_boxes : None or list of imgaug.augmntables.bbs.BoundingBox, optional List of bounding boxes on the image. If ``None``, the instance's bounding boxes will be copied. shape : tuple of int, optional The shape of the image on which the bounding boxes are placed. If ``None``, the instance's shape will be copied. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Shallow copy. Nrr4rNrrrrrr|s   zBoundingBoxesOnImage.copycCs4|dur dd|jD}|durt|j}t||S)aWCreate a deep copy of the ``BoundingBoxesOnImage`` object. Parameters ---------- bounding_boxes : None or list of imgaug.augmntables.bbs.BoundingBox, optional List of bounding boxes on the image. If ``None``, the instance's bounding boxes will be copied. shape : tuple of int, optional The shape of the image on which the bounding boxes are placed. If ``None``, the instance's shape will be copied. Returns ------- imgaug.augmentables.bbs.BoundingBoxesOnImage Deep copy. NcSrr)r;rrrrr/rz1BoundingBoxesOnImage.deepcopy..rrrrrr;s   zBoundingBoxesOnImage.deepcopycCr)zGet the bounding box(es) with given indices. Added in 0.4.0. Returns ------- list of imgaug.augmentables.bbs.BoundingBoxes Bounding box(es) with given indices. rrrrrr6rz BoundingBoxesOnImage.__getitem__cCr)a/Iterate over the bounding boxes in this container. Added in 0.4.0. Yields ------ BoundingBox A bounding box in this container. The order is identical to the order in the bounding box list provided upon class initialization. )rrr$rrrrCs zBoundingBoxesOnImage.__iter__cCr)zGet the number of items in this instance. Added in 0.4.0. Returns ------- int Number of items in this instance. )rrr$rrr__len__RrzBoundingBoxesOnImage.__len__cCrrrr$rrrr_rzBoundingBoxesOnImage.__repr__cCsdt|j|jfS)Nz"BoundingBoxesOnImage(%s, shape=%s))strrrNr$rrrrbszBoundingBoxesOnImage.__str__rrrr)NN),rrrrrrrsetterr+r-rrrrrrrrrrrrrzrrrrrrrarcr^rhrlrrrr|r;rrrrrrrrrrKs`       + #2 5   - $    rc@s\eZdZ   ddd Zd d Zd d ZeddZddZddZ eddZ ddZ dS)ryrmNrrQFrproc Cs4||_||_||_||_||_||_||_||_dSrrq) rrrrsrtrurvrwr+rxrrrrks z_LabelOnImageDrawer.__init__c Cs|jr ||||\}}|||\}}}}||ks!||kr#|S||j|||||jd|j|||j} | || ||||}|S)Nr) rw_do_raise_if_out_of_image_preprocess_colors_compute_bg_corner_coords_draw_label_arrrrNrrx_blend_label_arr_with_image_) rrM bounding_boxrsrtrrrr label_arrrrrr{ws   z"_LabelOnImageDrawer.draw_on_image_cCs|t||Sr)r{rr|)rrMrrrrrzsz!_LabelOnImageDrawer.draw_on_imagecCs.||rtd|j|j|j|j|jfdS)Nr~)rYrrrrrrN)rrMrrrrrs z-_LabelOnImageDrawer._do_raise_if_out_of_imagecCs|jdur t|jnd}|j}|jdurt|}n |dus#Jd|}|j}|jdur6t|}||fSd|dd|dd|d}tjd|d krQdnd tjd }||fS) NzGExpected `color` to be set when `color_bg` is not set, but it was None.gA`"?rgbX9?rgv/?r)rrnr)rrrrrtrsfull)rrrrtrsgrayrrrrs,       z&_LabelOnImageDrawer._preprocess_colorsc Cs|}|j}|jdd\}}|j|j|j}}} |d|j}||j} ||d}| |} t|| gd|d\}} t|| gd|d\}} ||| | fS)Nrrr)rurNr'r%r(r+rrb) rrMrroffsetr+r-rrrrrrrrs   z-_LabelOnImageDrawer._compute_bg_corner_coordsc Cs@tj|||f|d} |d| d<tj| ddt|||d} | S)Nr)rrr.r)r5r6textrrru)rrrrZ draw_textr) rrr+r-Z nb_channelsrrsrtrxrrrrrsz#_LabelOnImageDrawer._draw_label_arrc Cs|j}|dkr||||||ddf<|S|j}|tj} |||||ddftj} d|| || } t| dd|} | |||||ddf<|S)Nrrrrn)rvrrrZfloat64rb) rrMrrrrrrvr foreground backgroundblendrrrrs "z0_LabelOnImageDrawer._blend_label_arr_with_image_)rmNNrrQFrpro) rrrrr{rzrrrrrrrrrrryhs      ry)r __future__rrrr|numpyrZ skimage.drawrZskimage.measurerrbaserutilsr r r r objectr rryrrrrs8  C#