\;j@dZddlmZmZmZddlZddlZddlZddl Z ddl m Z ddlZddlZddlmZddlmZddlmZdd lmZmZmZmZmZd ZGd d eZ Gd deZ! ddZ"GddeZ#GddeZ$y)zClasses dealing with polygons.)print_functiondivisionabsolute_importN)imgaug)random) IAugmentable)normalize_shapeinterpolate_points_remove_out_of_image_fraction_project_coords__normalize_shift_argsc$d}t|tsd}|g}|g}t|D]`\}}t|jD]C\}}|j |j ||j||} | j |_Eb|s|dS|S)a?Apply a polygon recoverer to input polygons in-place. Parameters ---------- psois : list of imgaug.augmentables.polys.PolygonsOnImage or imgaug.augmentables.polys.PolygonsOnImage The possibly broken polygons, e.g. after augmentation. The `recoverer` is applied to them. psois_orig : list of imgaug.augmentables.polys.PolygonsOnImage or imgaug.augmentables.polys.PolygonsOnImage Original polygons that were later changed to `psois`. They are an extra input to `recoverer`. recoverer : imgaug.augmentables.polys._ConcavePolygonRecoverer The polygon recoverer used to repair broken input polygons. random_state : None or int or RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState An RNG to use during the polygon recovery. Returns ------- list of imgaug.augmentables.polys.PolygonsOnImage or imgaug.augmentables.polys.PolygonsOnImage List of repaired polygons. Note that this is `psois`, which was changed in-place. TFr) isinstancelist enumeratepolygons recover_fromexterior) psois psois_orig recoverer random_stateinput_was_listipsoijpolygonpoly_recs bG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\imgaug/augmentables/polys.pyrecover_psois_r"s4N eT " \ U#4#DMM2JAw --  *Q-"8"8";H (00G 3$ Qx LceZdZdZd1dZedZedZedZedZ edZ ed Z ed Z ed Z ed Zd ZdZd2dZdZdZdZdZd3dZej0dddZdZd4dZd5dZ d6dZdZ d7dZdZ d Z!d!Z"d"Z#d8d#Z$d$Z%d%Z&d9d&Z'e(d1d'Z) d:d(Z* d:d)Z+d:d*Z,d;d+Z-d;d,Z.d-Z/d.Z0d/Z1d0Z2y)<PolygonaClass representing polygons. Each polygon is parameterized by its corner points, given as absolute x- and y-coordinates with sub-pixel accuracy. Parameters ---------- exterior : list of imgaug.augmentables.kps.Keypoint or list of tuple of float or (N,2) ndarray List of points defining the polygon. May be either a ``list`` of :class:`~imgaug.augmentables.kps.Keypoint` objects or a ``list`` of ``tuple`` s in xy-form or a numpy array of shape (N,2) for ``N`` points in xy-form. All coordinates are expected to be the absolute subpixel-coordinates on the image, given as ``float`` s, e.g. ``x=10.7`` and ``y=3.4`` for a point at coordinates ``(10.7, 3.4)``. Their order is expected to be clock-wise. They are expected to not be closed (i.e. first and last coordinate differ). label : None or str, optional Label of the polygon, e.g. a string representing the class. NcVddlm}t|tr|s+t j dtj |_nt|d|r?t j |Dcgc]}|j|jgc}|_nt j |Dcgc] }|d|dgc}|_nntj|s Jd||jdk(r|jddk(sJd|jt j ||_t|jdk\xr8t j|jdd d f|jd d d f}|r|jd d |_||_y cc}wcc}w) zCreate a new Polygon instance.rKeypointrrdtyper zMExpected exterior to be a list of tuples (x, y) or an (N, 2) array, got type rzZExpected exterior to be a list of tuples (x, y) or an (N, 2) array, got an array of shape N)imgaug.augmentables.kpsr(rrnpzerosfloat32rxyia is_np_arrayndimshapelenallcloselabel)selfrr9r(point is_closeds r!__init__zPolygon.__init__es 5 h %!#rzz B HQK2 " 9A,C9A.3WWegg,>9A,C!D !# 9A,C9A.31XuQx,@9A,C!D >>(+ >2:= >+==A%(..*;q*@ &NN% &@ JJx0DM  ! # G DMM!Q$/r1u1EF   MM#2.DM 9,C ,Cs &F!%F&c|jS)a Alias for attribute ``exterior``. Added in 0.4.0. Returns ------- ndarray An ``(N, 2)`` ``float32`` ndarray containing the coordinates of this polygon. This identical to the attribute ``exterior``. rr:s r!coordszPolygon.coordss}}r#c(|jdddfS)zGet the x-coordinates of all points on the exterior. Returns ------- (N,2) ndarray ``float32`` x-coordinates array of all points on the exterior. Nrr?r@s r!xxz Polygon.xx}}QT""r#c(|jdddfS)zGet the y-coordinates of all points on the exterior. Returns ------- (N,2) ndarray ``float32`` y-coordinates array of all points on the exterior. Nr r?r@s r!yyz Polygon.yyrDr#cftjtj|jS)aGet the discretized x-coordinates of all points on the exterior. The conversion from ``float32`` coordinates to ``int32`` is done by first rounding the coordinates to the closest integer and then removing everything after the decimal point. Returns ------- (N,2) ndarray ``int32`` x-coordinates of all points on the exterior. )r.int32roundrCr@s r!xx_intzPolygon.xx_int xx)**r#cftjtj|jS)aGet the discretized y-coordinates of all points on the exterior. The conversion from ``float32`` coordinates to ``int32`` is done by first rounding the coordinates to the closest integer and then removing everything after the decimal point. Returns ------- (N,2) ndarray ``int32`` y-coordinates of all points on the exterior. )r.rHrIrFr@s r!yy_intzPolygon.yy_intrKr#cht|jdkry|jjS)aEstimate whether the polygon has a valid geometry. To to be considered valid, the polygon must be made up of at least ``3`` points and have a concave shape, i.e. line segments may not intersect or overlap. Multiple consecutive points are allowed to have the same coordinates. Returns ------- bool ``True`` if polygon has at least ``3`` points and is concave, otherwise ``False``. F)r7rto_shapely_polygonis_validr@s r!rQzPolygon.is_valids- t}}  !&&(111r#clt|jdkry|j}|jS)z{Compute the area of the polygon. Returns ------- number Area of the polygon. rOg)r7rrParear:polys r!rSz Polygon.areas0 t}}  !&&(yyr#cH|j}t|t|z S)aCompute the height of a bounding box encapsulating the polygon. The height is computed based on the two exterior coordinates with lowest and largest x-coordinates. Returns ------- number Height of the polygon. )rFmaxmin)r:rFs r!heightzPolygon.height WW2wR  r#cH|j}t|t|z S)aCompute the width of a bounding box encapsulating the polygon. The width is computed based on the two exterior coordinates with lowest and largest x-coordinates. Returns ------- number Width of the polygon. )rCrWrX)r:rCs r!widthz Polygon.width rZr#c>t|j|||_|S)ayProject the polygon onto an image with different shape in-place. The relative coordinates of all points remain the same. E.g. a point at ``(x=20, y=20)`` on an image ``(width=100, height=200)`` will be projected on a new image ``(width=200, height=100)`` to ``(x=40, y=10)``. 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 Shape of the original image. (Before resize.) to_shape : tuple of int Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.polys.Polygon Polygon object with new coordinates. The object may have been modified in-place. )rrArr: from_shapeto_shapes r!project_zPolygon.project_s8( ZJ  r#cB|jj||S)aProject the polygon onto an image with different shape. The relative coordinates of all points remain the same. E.g. a point at ``(x=20, y=20)`` on an image ``(width=100, height=200)`` will be projected on a new image ``(width=200, height=100)`` to ``(x=40, y=10)``. 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 Shape of the original image. (Before resize.) to_shape : tuple of int Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.polys.Polygon Polygon object with new coordinates. )deepcopyrar^s r!projectzPolygon.project9s2}}'' H==r#ct|jdkDsJdg}|jD]'\}}||z dz||z dzz}|j|)tj|}tj |}|r|||fS|S)a)Find the index of the exterior point closest to given coordinates. "Closeness" is here defined based on euclidean distance. This method will raise an ``AssertionError`` if the exterior contains no points. Parameters ---------- x : number X-coordinate around which to search for close points. y : number Y-coordinate around which to search for close points. return_distance : bool, optional Whether to also return the distance of the closest point. Returns ------- int Index of the closest point. number Euclidean distance to the closest point. This value is only returned if `return_distance` was set to ``True``. rzOCannot find the closest point on a polygon which's exterior contains no points.r)r7rappendr.sqrtargmin) r:r1r2return_distance distancesx2y2dist closest_idxs r!find_closest_point_indexz Polygon.find_closest_point_indexTs:4==!A% # " #% mmFBFq=BFq=0D   T "$GGI& ii *  + 66 6r#c|j|}t|dk(r |jS|jt|Dcgc]}|jc}z Scc}w)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``. r)clip_out_of_imager7rSsum)r:image polys_clippedrUs r!compute_out_of_image_areaz!Polygon.compute_out_of_image_area~sU(..u5 }  "99 yy3mDmd mDEEEDsA c|j}|dk(r|jj|S|j||z S)aCompute fraction of polygon area outside of the image plane. This estimates ``f = A_ooi / A``, where ``A_ooi`` is the area of the polygon 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 polygon area that is outside of the image plane. Returns ``0.0`` if the polygon is fully inside of the image plane or has zero points. If the polygon has an area of zero, the polygon is treated similarly to a :class:`LineString`, i.e. the fraction of the line that is outside the image plane is returned. r)rSto_line_stringcompute_out_of_image_fractionru)r:rsrSs r!rxz%Polygon.compute_out_of_image_fractionsD8yy 19&&(FFuM M--e4t;;r#c,|j|dd S)a Estimate whether the polygon is fully inside an image plane. 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 ``int`` s. Returns ------- bool ``True`` if the polygon is fully inside the image area. ``False`` otherwise. Tfullypartlyis_out_of_imager:rss r!is_fully_within_imagezPolygon.is_fully_within_images$''T$'GGGr#c,|j|dd S)aEstimate whether the polygon is at least partially inside an image. 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 ``int`` s. Returns ------- bool ``True`` if the polygon is at least partially inside the image area. ``False`` otherwise. TFrzr}rs r!is_partly_within_imagezPolygon.is_partly_within_images$''T%'HHHr#ct|jdk(r td|j}|j |ry|j |ddr|S|j |}t|dkDr|S|S)agEstimate whether the polygon is partially/fully outside of an image. 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 ``int`` s. fully : bool, optional Whether to return ``True`` if the polygon is fully outside of the image area. partly : bool, optional Whether to return ``True`` if the polygon is at least partially outside fo the image area. Returns ------- bool ``True`` if the polygon is partially/fully outside of the image area, depending on defined parameters. ``False`` otherwise. rzXCannot determine whether the polygon is inside the image, because it contains no points.FTrz)r7r Exceptionrwrr~rq)r:rsr{r|lspolyss r!r~zPolygon.is_out_of_images: t}}  "HI I " # #E *  e5  >M &&u- u:>M r#zPolygon.clip_out_of_image()z3clip_out_of_image() has the exactly same interface.)alt_funccommentc$|j|S)z>Cut off all parts of the polygon that are outside of an image.)rqrs r!cut_out_of_imagezPolygon.cut_out_of_images %%e,,r#cddl}t|jdk(rgSt|jdvrc|jd}|j |}t|dksJt|dk(rgS|j |dj gStj|r|jddn|dd\}}|j}|jjd |df||fd|fg}|j|} |jj|jj|jj j"|jj$f} t'| |jjr|jj)| g} nt'| |jj(rnt'| | r|jj)g} nlt'| |jj*r| j,sJgSt/| ||jt1d t3| |||jfzg} | j4D]2} | j7tj9| |j: 4g} | D]b}d}d}|jD](\}}|j=||d \}}|||ks%|}|}*|A|j?|}| j7|d| S)aFCut off all parts of the polygon that are outside of an image. This operation may lead to new points being created. As a single polygon may be split into multiple new polygons, the result is always a list, which may contain more than one output polygon. This operation will return an empty list if the polygon is completely outside of the image plane. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use for the clipping of the polygon. 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 ``int`` s. Returns ------- list of imgaug.augmentables.polys.Polygon Polygon, clipped to fall within the image dimensions. Returned as a ``list``, because the clipping can split the polygon into multiple parts. The list may also be empty, if the polygon was fully outside of the image plane. rN)r rFclosedr r?rrrz}Got an unexpected result of type %s from Shapely for image (%d, %d) and polygon %s. This is an internal error. Please report.r9Tr1r2ri) shapely.geometryr7rrwrqrcrAr3r4r6rPgeometryr% intersection LineStringMultiLineStringr;Point MultiPointr MultiPolygonGeometryCollectionis_emptyprintrtypegeomsrf from_shapelyr9rochange_first_point_by_index)r:rsshapelyr ls_clippedhw poly_shapely poly_imagemultipoly_inter_shapely ignore_typesrpoly_inter_shapelypolygons_reorderedrbest_idx best_distr1r2 point_idxrmpolygon_reordereds r!rqzPolygon.clip_out_of_image#s8  t}}  "I t}}  '$$E$2B--e4Jz?a' '':!# MM:a=+?+?M@A A#%>>%#8u{{1QeAaj1..0 %%--v1v1v1v.NO ".";";J"G((33((88((..44((335  -w/?/?/G/G H&-&6&6&C&C()'+ # /((557  / >'.&6&6&C&CB&G # /((;;=+33 33I )5$-- @!011a$HH "9"?"?  OOG001C7;zz1C D#@ GHI 1")"B"B1d#C#4 4#ti'7(H $I & #77A""))*;< "!r#cr|jdddfxx|z cc<|jdddfxx|z cc<|S)atMove this polygon 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.polys.Polygon Shifted polygon. The object may have been modified in-place. Nrr r?)r:r1r2s r!shift_zPolygon.shift_s90 adq  adq  r#clt||||||\}}|jj||S)aMove this polygon 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.polys.Polygon Shifted polygon. toprightbottomleftr1r2rrcrr:r1r2rrrrs r!shiftz Polygon.shift=R% qcvDB1}}%%Q%//r#cd}d}|d||d||d| ||tj|}||tj|dz}||tj|dz}|||dz}|||}|| |} || | } || | dz} |jdk(r-tj|sJd t |d |g}n7|jdk(r(tj|r|g|j d z}|d krd }n|dkDrd}| r6|j|r%tdt |d|j d |j}|jtj}tjj|j|j |j \}}t#|d kDrT|dk(rtj||||f<n4|d k(rn.d|z |||ddfz|tj|zz|||f<|j%d}|j%d}|j'|||| | }|j)||| | | }|j*tj,k(r;tj.tj0|d dj|}|S|j|}|S)aDraw the polygon on an image. Parameters ---------- image : (H,W,C) ndarray The image onto which to draw the polygon. Usually expected to be of dtype ``uint8``, though other dtypes are also handled. color : iterable of int, optional The color to use for the whole polygon. Must correspond to the channel layout of the image. Usually RGB. The values for `color_face`, `color_lines` and `color_points` will be derived from this color if they are set to ``None``. This argument has no effect if `color_face`, `color_lines` and `color_points` are all set anything other than ``None``. color_face : None or iterable of int, optional The color to use for the inner polygon area (excluding perimeter). Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 1.0``. color_lines : None or iterable of int, optional The color to use for the line (aka perimeter/border) of the polygon. Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 0.5``. color_points : None or iterable of int, optional The color to use for the corner points of the polygon. Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 0.5``. alpha : float, optional The opacity of the whole polygon, where ``1.0`` denotes a completely visible polygon and ``0.0`` an invisible one. The values for `alpha_face`, `alpha_lines` and `alpha_points` will be derived from this alpha value if they are set to ``None``. This argument has no effect if `alpha_face`, `alpha_lines` and `alpha_points` are all set anything other than ``None``. alpha_face : None or number, optional The opacity of the polygon's inner area (excluding the perimeter), where ``1.0`` denotes a completely visible inner area and ``0.0`` an invisible one. If this is ``None``, it will be derived from ``alpha * 0.5``. alpha_lines : None or number, optional The opacity of the polygon's line (aka perimeter/border), where ``1.0`` denotes a completely visible line and ``0.0`` an invisible one. If this is ``None``, it will be derived from ``alpha * 1.0``. alpha_points : None or number, optional The opacity of the polygon's corner points, where ``1.0`` denotes completely visible corners and ``0.0`` invisible ones. If this is ``None``, it will be derived from ``alpha * 1.0``. size : int, optional Size of the polygon. The sizes of the line and points are derived from this value, unless they are set. size_lines : None or int, optional Thickness of the polygon's line (aka perimeter/border). If ``None``, this value is derived from `size`. size_points : int, optional Size of the points in pixels. If ``None``, this value is derived from ``3 * size``. raise_if_out_of_image : bool, optional Whether to raise an error if the polygon 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 Image with the polygon drawn on it. Result dtype is the same as the input dtype. c6|Jd|dt|dy)Nz Expected 'z' to not be None, got type .)r)arg_name arg_values r!_assert_not_nonez/Polygon.draw_on_image.._assert_not_none2s#( 1d9o0 1(r#c||S|SN)vardefaults r! _default_toz*Polygon.draw_on_image.._default_to7s{Jr#coloralphasizeg?rOrzJGot a 2D image. Expected then 'color_face' to be a single number, but got rr,{Gz?rgGz?r zCannot draw polygon z on image with shape r6NFrT)rrrraise_if_out_of_image)r.arrayr5r3is_single_numberstrr6r~rr+astyper0skimagedrawrrMrJr7rwdraw_lines_on_imagedraw_points_on_imageruint8cliprI)r:rsr color_face color_lines color_pointsr alpha_face alpha_lines alpha_pointsr size_lines size_pointsrrr input_dtyperesultrrccls_open ls_closeds r! draw_on_imagezPolygon.draw_on_imagesv 1   %(%(&!RXXe_= !+rxx/DE "<%31FG  US[9 !+u5 "<7  T2 !+tax8 ::?&&z2 <),Z; <2%J ZZ1_!4!4Z!@$ B7J  J $ J T%9%9%%@+.t9ekkCD Dkk bjj)%% KKEKK&9B r7Q;Q!#J!7r2vq^vb"ai'88 2::j#99:r2v %%U%3''t'4 .. +[3H/J-- ,l4I.K   rxx 'WWRXXf-q#6==kJF ]];/F r#c.|jdvsJd|jdt|jdkr t d|j }|j |}|j|ddr|S|j}|j}|tj|z }|tj|z }tj|}tj|} tjj|||| f \} } tj || ftj" } d| | | f<|jd k(r>tj$| d d d d tj&fd d |jdf} || zS)aExtract all image pixels within the polygon area. This method returns a rectangular image array. All pixels within that rectangle that do not belong to the polygon area will be filled with zeros (i.e. they will be black). The method will also zero-pad the image if the polygon 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 polygon. Returns ------- (H',W') ndarray or (H',W',C) ndarray Pixels within the polygon. Zero-padded if the polygon is partially/fully outside of the image. )rrOz-Expected image of shape (H,W,[C]), got shape rrzOPolygon must be made up of at least 3 points to extract its area from an image.TFrzrr*rONr )r5r6r7rrto_bounding_boxextract_from_imager~rJrMr.rXrWrrrr/booltilenewaxis) r:rsbbbb_arearCrFxx_maskyy_mask height_mask width_maskrr_facecc_facemasks r!rzPolygon.extract_from_imagesn*zzV#    # t}}  ">? ? ! ! #''.   T%  @N [[ [[rvvbz/rvvbz/ffWo VVG_ "<<// W[*$=0?xxj1A!%Wg  ::?7741bjj 01Aq%++a.3IJD~r#ct|jdk(r td|j||d\}}|A||kDr<|s|j S|j|ddf}td|d|d|fz|j |S)a4 Reorder exterior points so that the point closest to given x/y is first. This method takes a given ``(x,y)`` coordinate, finds the closest corner point on the exterior and reorders all exterior corner points so that the found point becomes the first one in the array. If no matching points are found, an exception is raised. Parameters ---------- x : number X-coordinate of the point. y : number Y-coordinate of the point. max_distance : None or number, optional Maximum distance past which possible matches are ignored. If ``None`` the distance limit is deactivated. raise_if_too_far_away : bool, optional Whether to raise an exception if the closest found point is too far away (``True``) or simply return an unchanged copy if this object (``False``). Returns ------- imgaug.augmentables.polys.Polygon Copy of this polygon with the new point order. rz=Cannot reorder polygon points, because it contains no points.TrNzFClosest found point (%.9f, %.9f) exceeds max_distance of %.9f exceededr )r7rrrorcr)r:r1r2 max_distanceraise_if_too_far_awayrn closest_dist closest_points r!change_first_point_by_coordsz$Polygon.change_first_point_by_coordssD t}}  "23 3%)$A$A1d%B%,! \  # |(C(}}& MM+q.9M !!$mA&6 #FFG G// <==? ">> ]]9:q= )4==)Q+G H }}h}//r#ct|jdk(r|S|jd}|j|}|jdd}||_|S)aDerive a new poly with ``N`` interpolated points per edge in-place. See :func:`~imgaug.augmentables.lines.LineString.subdivide` for details. Added in 0.4.0. Parameters ---------- points_per_edge : int Number of points to interpolate on each edge. Returns ------- imgaug.augmentables.polys.Polygon Polygon with subdivided edges. The object may have been modified in-place. r TrNr,)r7rrw subdividerA)r:points_per_edgerls_subexterior_subdivideds r! subdivide_zPolygon.subdivide_sX& t}}  "K   -o.$mmCR0+  r#c@|jj|S)aDerive a new polygon with ``N`` interpolated points per edge. See :func:`~imgaug.augmentables.lines.LineString.subdivide` for details. Added in 0.4.0. Parameters ---------- points_per_edge : int Number of points to interpolate on each edge. Returns ------- imgaug.augmentables.polys.Polygon Polygon with subdivided edges. rcrr:rs r!rzPolygon.subdivide/s$}}))/::r#cddl}|jj|jDcgc] }|d|dfc}Scc}w)zConvert this polygon to a ``Shapely`` ``Polygon``. Returns ------- shapely.geometry.Polygon The ``Shapely`` ``Polygon`` matching this polygon's exterior. rNr )rrr%r)r:rr;s r!rPzPolygon.to_shapely_polygonCsF ''/3}} =}eeAha !} =? ? =sAc2t|j||S)aJConvert this polygon to a ``Shapely`` ``LineString`` object. Parameters ---------- closed : bool, optional Whether to return the line string with the last point being identical to the first point. interpolate : int, optional Number of points to interpolate between any pair of two consecutive points. These points are added to the final line string. Returns ------- shapely.geometry.LineString The ``Shapely`` ``LineString`` matching the polygon's exterior. )r interpolate)&_convert_points_to_shapely_line_stringr)r:rr s r!to_shapely_line_stringzPolygon.to_shapely_line_stringRs&6 MM&kC Cr#cddlm}|j}|j}|t |t |t |t ||j S)zConvert this polygon to a bounding box containing the polygon. Returns ------- imgaug.augmentables.bbs.BoundingBox Bounding box that tightly encapsulates the polygon. r) BoundingBox)x1rky1rlr9)imgaug.augmentables.bbsrrCrFrXrWr9)r:rrCrFs r!rzPolygon.to_bounding_boxhsF 8 WW WWc"g#b'!"g#b'!%- -r#chddlm}|jDcgc]}||d|dc}Scc}w)zConvert this polygon's exterior to ``Keypoint`` instances. Returns ------- list of imgaug.augmentables.kps.Keypoint Exterior vertices as :class:`~imgaug.augmentables.kps.Keypoint` instances. rr'r r)r-r(r)r:r(r;s r! to_keypointszPolygon.to_keypointszs2 5>BmmLmU58uQx0mLLLs/c ddlm}|rt|jdkr||j|jS|t j |j|jddddfgd|jS)a Convert this polygon's exterior to a ``LineString`` instance. Parameters ---------- closed : bool, optional Whether to close the line string, i.e. to add the first point of the `exterior` also as the last point at the end of the line string. This has no effect if the polygon has a single point or zero points. Returns ------- imgaug.augmentables.lines.LineString Exterior of the polygon as a line string. r)rr rNr)imgaug.augmentables.linesrr7rr9r.r)r:rrs r!rwzPolygon.to_line_stringsk" 9T]]+q0dmm4::> > NNDMM4==1a+@A J** r#cddl}t||jjsJdt |d|j duxs"t |j jdk(}|r tg|Stj|j jDcgc] \}}||g c}}}t||Scc}}w)aCreate a polygon from a ``Shapely`` ``Polygon``. .. note:: This will remove any holes in the shapely polygon. Parameters ---------- polygon_shapely : shapely.geometry.Polygon The shapely polygon. label : None or str, optional The label of the new polygon. Returns ------- imgaug.augmentables.polys.Polygon A polygon with the same exterior as the ``Shapely`` ``Polygon``. rNzAExpected the input to be a shapely.geometry.Polgon instance. Got rr) rrrr%rrr7rAr.r0)polygon_shapelyr9rhas_no_exteriorr1r2rs r!rzPolygon.from_shapelys. /7+;+;+C+CD 2o. 1 2D  $ $ , 9?++223q8  2U+ +::"1":":"A"AC"A$*Aq!"1v"ACDxu--Cs"C c*|j|||S)aAlias for :func:`Polygon.exterior_almost_equals`. Parameters ---------- other : imgaug.augmentables.polys.Polygon or (N,2) ndarray or list of tuple See :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals`. max_distance : number, optional See :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals`. points_per_edge : int, optional See :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals`. Returns ------- bool Whether the two polygon's exteriors can be viewed as equal (approximate test). rr)exterior_almost_equalsr:otherrrs r!coords_almost_equalszPolygon.coords_almost_equalss%2**  o+O Or#cRt|trttj|}nEt j |r t|}n$t|tsJdt|d|jdj|jd||S)aEstimate if this and another polygon's exterior are almost identical. The two exteriors can have different numbers of points, but any point randomly sampled on the exterior of one polygon should be close to the closest point on the exterior of the other polygon. .. note:: This method works in an approximative way. One can come up with polygons with fairly different shapes that will still be estimated as equal by this method. In practice however this should be unlikely to be the case. The probability for something like that goes down as the interpolation parameter is increased. Parameters ---------- other : imgaug.augmentables.polys.Polygon or (N,2) ndarray or list of tuple The other polygon with which to compare the exterior. If this is an ``ndarray``, it is assumed to represent an exterior. It must then have dtype ``float32`` and shape ``(N,2)`` with the second dimension denoting xy-coordinates. If this is a ``list`` of ``tuple`` s, it is assumed to represent an exterior. Each tuple then must contain exactly two ``number`` s, denoting xy-coordinates. max_distance : number, optional The maximum euclidean distance between a point on one polygon and the closest point on the other polygon. If the distance is exceeded for any such pair, the two exteriors are not viewed as equal. The points are either the points contained in the polygon's exterior ndarray or interpolated points between these. points_per_edge : int, optional How many points to interpolate on each edge. Returns ------- bool Whether the two polygon's exteriors can be viewed as equal (approximate test). z_Expected 'other' to be a list of coordinates, a coordinate array or a single Polygon. Got type rTrr) rrr%r.r0r3r4rrwrrs r!rzPolygon.exterior_almost_equalssX eT "BJJu-.E ^^E "ENEeW- L=A%[K L-""$"/DD   -%+E  r#c^|j|jk7ry|j|||S)aS Estimate if this polygon's and another's geometry/labels are similar. This is the same as :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals` but additionally compares the labels. Parameters ---------- other : imgaug.augmentables.polys.Polygon The other object to compare against. Expected to be a ``Polygon``. max_distance : float, optional See :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals`. points_per_edge : int, optional See :func:`~imgaug.augmentables.polys.Polygon.exterior_almost_equals`. Returns ------- bool ``True`` if the coordinates are almost equal and additionally the labels are equal. Otherwise ``False``. Fr)r9rrs r! almost_equalszPolygon.almost_equals s88 :: $**  o+O Or#c(|j||S)a3Create a shallow copy of this object. Parameters ---------- exterior : list of imgaug.augmentables.kps.Keypoint or list of tuple or (N,2) ndarray, optional List of points defining the polygon. See :func:`~imgaug.augmentables.polys.Polygon.__init__` for details. label : None or str, optional If not ``None``, the ``label`` of the copied object will be set to this value. Returns ------- imgaug.augmentables.polys.Polygon Shallow copy. rr9)rcr:rr9s r!copyz Polygon.copyAs&}}he}<>6(TF2/09 vOB=*:. & #8r#r%ceZdZdZdZedZejdZedZdZ dZ dd Z d d Z d d Z d Zd ZdZdZd!dZd"dZdZdZdZdZdZdZd#dZd#dZdZdZdZdZdZ y)$PolygonsOnImageaContainer for all polygons on a single image. Parameters ---------- polygons : list of imgaug.augmentables.polys.Polygon List of polygons 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.polys import Polygon, PolygonsOnImage >>> image = np.zeros((100, 100)) >>> polys = [ >>> Polygon([(0.5, 0.5), (100.5, 0.5), (100.5, 100.5), (0.5, 100.5)]), >>> Polygon([(50.5, 0.5), (100.5, 50.5), (50.5, 100.5), (0.5, 50.5)]) >>> ] >>> polys_oi = PolygonsOnImage(polys, shape=image.shape) c2||_t||_yr)rr r6r:rr6s r!r=zPolygonsOnImage.__init__s  $U+ r#c|jS)zGet the polygons in this container. Added in 0.4.0. Returns ------- list of Polygon Polygons within this container. rr@s r!itemszPolygonsOnImage.itemss}}r#c||_y)zSet the polygons in this container. Added in 0.4.0. Parameters ---------- value : list of Polygon Polygons within this container. NrM)r:values r!rNzPolygonsOnImage.itemss  r#c2t|jdk(S)zEstimate whether this object contains zero polygons. Returns ------- bool ``True`` if this object contains zero polygons. r)r7rr@s r!emptyzPolygonsOnImage.emptys4==!Q&&r#ct|}|dd|jddk(r ||_|St|jD].\}}|j |j||j |<0||_|S)aProject all polygons from one image shape to a new one in-place. Added in 0.4.0. Parameters ---------- image : ndarray or tuple of int New image onto which the polygons are to be projected. May also simply be that new image's shape ``tuple``. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Object containing all projected polygons. The object and its items may have been modified in-place. rr)r r6rrNrar)r:rson_shaperitems r!on_zPolygonsOnImage.on_sr&#5) Aa=DJJqO +!DJK ,GAt#}}TZZBDMM! -  r#c@|jj|S)aProject all polygons from one image shape to a new one. Parameters ---------- image : ndarray or tuple of int New image onto which the polygons are to be projected. May also simply be that new image's shape ``tuple``. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Object containing all projected polygons. )rcrVrs r!onzPolygonsOnImage.ons }}""5))r#Ncd|jD] }|j||||||||| | | | |  }"|S)aQDraw all polygons onto a given image. Parameters ---------- image : (H,W,C) ndarray The image onto which to draw the bounding boxes. This image should usually have the same shape as set in ``PolygonsOnImage.shape``. color : iterable of int, optional The color to use for the whole polygons. Must correspond to the channel layout of the image. Usually RGB. The values for `color_face`, `color_lines` and `color_points` will be derived from this color if they are set to ``None``. This argument has no effect if `color_face`, `color_lines` and `color_points` are all set anything other than ``None``. color_face : None or iterable of int, optional The color to use for the inner polygon areas (excluding perimeters). Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 1.0``. color_lines : None or iterable of int, optional The color to use for the lines (aka perimeters/borders) of the polygons. Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 0.5``. color_points : None or iterable of int, optional The color to use for the corner points of the polygons. Must correspond to the channel layout of the image. Usually RGB. If this is ``None``, it will be derived from ``color * 0.5``. alpha : float, optional The opacity of the whole polygons, where ``1.0`` denotes completely visible polygons and ``0.0`` invisible ones. The values for `alpha_face`, `alpha_lines` and `alpha_points` will be derived from this alpha value if they are set to ``None``. This argument has no effect if `alpha_face`, `alpha_lines` and `alpha_points` are all set anything other than ``None``. alpha_face : None or number, optional The opacity of the polygon's inner areas (excluding the perimeters), where ``1.0`` denotes completely visible inner areas and ``0.0`` invisible ones. If this is ``None``, it will be derived from ``alpha * 0.5``. alpha_lines : None or number, optional The opacity of the polygon's lines (aka perimeters/borders), where ``1.0`` denotes completely visible perimeters and ``0.0`` invisible ones. If this is ``None``, it will be derived from ``alpha * 1.0``. alpha_points : None or number, optional The opacity of the polygon's corner points, where ``1.0`` denotes completely visible corners and ``0.0`` invisible ones. Currently this is an on/off choice, i.e. only ``0.0`` or ``1.0`` are allowed. If this is ``None``, it will be derived from ``alpha * 1.0``. size : int, optional Size of the polygons. The sizes of the line and points are derived from this value, unless they are set. size_lines : None or int, optional Thickness of the polygon lines (aka perimeter/border). If ``None``, this value is derived from `size`. size_points : int, optional The size of all corner points. If set to ``C``, each corner point will be drawn as a square of size ``C x C``. raise_if_out_of_image : bool, optional Whether to raise an error if any polygon 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 Image with drawn polygons. ) rrrrrrrrrrrr)rr)r:rsrrrrrrrrrrrrrUs r!rzPolygonsOnImage.draw_on_image sVxMMD&&%')%')%'&;'E"  r#c|jDcgc]"}|j|j||s|$c}|_|Scc}w)aRemove all polygons that are fully/partially OOI in-place. 'OOI' is the abbreviation for 'out of image'. Added in 0.4.0. Parameters ---------- fully : bool, optional Whether to remove polygons that are fully outside of the image. partly : bool, optional Whether to remove polygons that are partially outside of the image. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Reduced set of polygons. Those that are fully/partially outside of the given image plane are removed. The object and its items may have been modified in-place. rz)rr~r6)r:r{r|rUs r!remove_out_of_image_z$PolygonsOnImage.remove_out_of_image_wsK0"]] *T'' %'O ]    s'AcB|jj||S)aRemove all polygons that are fully/partially outside of an image. Parameters ---------- fully : bool, optional Whether to remove polygons that are fully outside of the image. partly : bool, optional Whether to remove polygons that are partially outside of the image. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Reduced set of polygons. Those that are fully/partially outside of the given image plane are removed. )rcr[)r:r{r|s r!remove_out_of_imagez#PolygonsOnImage.remove_out_of_images$}}33E6BBr#ct||S)aRemove all Polys with an OOI fraction of ``>=fraction`` in-place. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a polygon has to have in order to be removed. A fraction of ``1.0`` removes only polygons that are ``100%`` outside of the image. A fraction of ``0.0`` removes all polygons. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Reduced set of polygons, 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 r:fractions r!remove_out_of_image_fraction_z-PolygonsOnImage.remove_out_of_image_fraction_s*.dH==r#c@|jj|S)aRemove all Polys with an out of image fraction of ``>=fraction``. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a polygon has to have in order to be removed. A fraction of ``1.0`` removes only polygons that are ``100%`` outside of the image. A fraction of ``0.0`` removes all polygons. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Reduced set of polygons, with those that had an out of image fraction greater or equal the given one removed. )r&rar_s r!remove_out_of_image_fractionz,PolygonsOnImage.remove_out_of_image_fractions(yy{88BBr#c|jDcgc]$}|j|jD]}|&c}}|_|Scc}}w)aClip off all parts from all polygons that are OOI in-place. 'OOI' is the abbreviation for 'out of image'. .. note:: The result can contain fewer polygons than the input did. That happens when a polygon is fully outside of the image plane. .. note:: The result can also contain *more* polygons than the input did. That happens when distinct parts of a polygon are only connected by areas that are outside of the image plane and hence will be clipped off, resulting in two or more unconnected polygon parts that are left in the image plane. Added in 0.4.0. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Polygons, clipped to fall within the image dimensions. The count of output polygons may differ from the input count. The object and its items may have been modified in-place. )rrqr6)r:rU poly_clippeds r!clip_out_of_image_z"PolygonsOnImage.clip_out_of_image_sR< D% $ 6 6tzz B   B %D  Ds)Ac>|jjS)aEClip off all parts from all polygons that are outside of an image. .. note:: The result can contain fewer polygons than the input did. That happens when a polygon is fully outside of the image plane. .. note:: The result can also contain *more* polygons than the input did. That happens when distinct parts of a polygon are only connected by areas that are outside of the image plane and hence will be clipped off, resulting in two or more unconnected polygon parts that are left in the image plane. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Polygons, clipped to fall within the image dimensions. The count of output polygons may differ from the input count. )r&rfr@s r!rqz!PolygonsOnImage.clip_out_of_images.yy{--//r#ct|jD]%\}}|j|||j|<'|S)aEMove the polygons 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.polys.PolygonsOnImage Shifted polygons. r)rrr)r:r1r2rrUs r!rzPolygonsOnImage.shift_s;.!/GAt#{{Q!{4DMM! 0 r#clt||||||\}}|jj||S)aMove the polygons 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.polys.PolygonsOnImage Shifted polygons. rrrrs r!rzPolygonsOnImage.shift+rr#c|t|jD]#\}}|j||j|<%|Sa?Interpolate ``N`` points on each polygon. Added in 0.4.0. Parameters ---------- points_per_edge : int Number of points to interpolate on each edge. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Subdivided polygons. )rrr)r:rrrUs r!rzPolygonsOnImage.subdivide_Xs7 !/GAt#?DMM! 0 r#c@|jj|Srkrrs r!rzPolygonsOnImage.subdividels }}))/::r#c|jr%tjdtjStj|j Dcgc]}|j c}Scc}w)zConvert all polygon coordinates to one array of shape ``(N,2)``. Added in 0.4.0. Returns ------- (N, 2) ndarray Array containing all xy-coordinates of all polygons within this instance. r)r*)rRr.r/r0rrrrTs r! to_xy_arrayzPolygonsOnImage.to_xy_array~sI ::88F"**5 5~~Gt}}GHHGsA)c ptj|tj}|jddk(s6|jdk(r|jddk(sJd|jdd}|j D]}t |j}||zt |ksIJdt |t|j Dcgc]}t |jc}fz||||z|jddd f<||z }|t |k(sJd t ||fz|Scc}w) aModify the corner coordinates of all polygons in-place. .. note:: This currently expects that `xy` contains exactly as many coordinates as the polygons within this instance have corner points. Otherwise, an ``AssertionError`` will be raised. .. warning:: This does not validate the new coordinates or repair the resulting polygons. If bad coordinates are provided, the result will be invalid polygons (e.g. self-intersections). Added in 0.4.0. Parameters ---------- xy : (N, 2) ndarray or iterable of iterable of number XY-Coordinates of ``N`` corner points. ``N`` must match the number of corner points in all polygons within this instance. Returns ------- PolygonsOnImage This instance itself, with updated coordinates. Note that the instance was modified in-place. r*rrr,z4Expected input array to have shape (N,2), got shape rzpReceived fewer points than there are corner points in the exteriors of all polygons. Got %d points, expected %d.N.zExpected to get exactly as many xy-coordinates as there are points in the exteriors of all polygons within this instance. Got %d points, could only assign %d points.) r.rr0r6r5rr7rrr)r:xycounterrU nb_pointsps r!fill_from_xy_array_z#PolygonsOnImage.fill_from_xy_array_sG<XXb +xx{aBGGqLRXXb\Q5F +!xx * +GMMDDMM*IY&#b'1 MIGS4==!I=a#ajj/=!IJLLL M1 %'wwy/@$ADMM!S& ! y G"#b'! $ :B=# # $!  "JsD3cddlm}|jr|g|jSt j |j Dcgc]}|jc}d}|j||jScc}w)aConvert the polygons to one ``KeypointsOnImage`` instance. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage A keypoints instance containing ``N`` coordinates for a total of ``N`` points in all exteriors of the polygons within this container. Order matches the order in ``polygons``. r )KeypointsOnImagerrr) rvrRr6r.rrr from_xy_array)r:rvrU exteriorss r!to_keypoints_on_imagez%PolygonsOnImage.to_keypoints_on_imagesk ' ::#Bdjj9 9NN'+}} 5}tT]]} 5  --itzz-JJ 6sA=c|j}|Dcgc]}|j}}t|Dcgc] }t|c}}t|j|k(sJd|t|jfz|j }d}|D]<}|j}|||t|zddf|ddddf<|t|z }>|j |_|Scc}wcc}w)a4Invert the output of ``to_keypoints_on_image()`` in-place. This function writes in-place into this ``PolygonsOnImage`` instance. Added in 0.4.0. Parameters ---------- kpsoi : imgaug.augmentables.kps.KeypointsOnImages Keypoints to convert back to polygons, i.e. the outputs of ``to_keypoints_on_image()``. Returns ------- PolygonsOnImage Polygons container with updated coordinates. Note that the instance is also updated in-place. z Expected %d coordinates, got %d.rN)rrrrr7 keypointsrnr6) r:kpsoirrUryr nb_points_expxy_arrrqs r!invert_to_keypoints_on_image_z-PolygonsOnImage.invert_to_keypoints_on_image_s* /45utT]]u 59E9xS]9EF 5??#}4 6 .s5??325 5 64""$D}}H#GGCM,A$A1$DEHQTN s8} $G[[  6Es C%C*cj||jdd}|t|j}t||S)aCreate a shallow copy of this object. Parameters ---------- polygons : None or list of imgaug.augmentables.polys.Polygons, optional List of polygons on the image. If not ``None``, then the ``polygons`` attribute of the copied object will be set to this value. shape : None or tuple of int or ndarray, optional 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. If not ``None``, then the ``shape`` attribute of the copied object will be set to this value. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Shallow copy. N)rtupler6rIrKs r!r&zPolygonsOnImage.copys8.  }}Q'H =$**%Ex//r#c|'|jDcgc]}|j}}|t|j}t ||Scc}w)aCreate a deep copy of this object. Parameters ---------- polygons : None or list of imgaug.augmentables.polys.Polygons, optional List of polygons on the image. If not ``None``, then the ``polygons`` attribute of the copied object will be set to this value. shape : None or tuple of int or ndarray, optional 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. If not ``None``, then the ``shape`` attribute of the copied object will be set to this value. Returns ------- imgaug.augmentables.polys.PolygonsOnImage Deep copy. )rrcrr6rI)r:rr6rUs r!rczPolygonsOnImage.deepcopy sN0  48MMBMD MHB =$**%Ex// CsA c |j|S)zGet the polygon(s) with given indices. Added in 0.4.0. Returns ------- list of imgaug.augmentables.polys.Polygon Polygon(s) with given indices. rMr)s r!r+zPolygonsOnImage.__getitem__@r,r#c,t|jS)aIterate over the polygons in this container. Added in 0.4.0. Yields ------ Polygon A polygon in this container. The order is identical to the order in the polygon list provided upon class initialization. )r.rr@s r!r/zPolygonsOnImage.__iter__MsDMM""r#c,t|jS)zGet the number of items in this instance. Added in 0.4.0. Returns ------- int Number of items in this instance. )r7rNr@s r!__len__zPolygonsOnImage.__len__\s4::r#c"|jSrr1r@s r!r3zPolygonsOnImage.__repr__ir4r#cNdt|jd|jdS)NzPolygonsOnImage(z, shape=))rrr6r@s r!r2zPolygonsOnImage.__str__ls   , ,r#r;r9rr:r@)!rArBrCrDr=rErNsetterrRrVrXrr[r]rarcrfrqrrrrrnrtrzrr&rcr+r/rr3r2rr#r!rIrIs2,   \\   ' ':*(5959,059;?,1l\:C(>.C, D026+0Z(;$I 6pK*$L0>0@ & # ,r#rIc.ddl}t|dkrtdt|fz|Dcgc] }|d|df}}|dkDr t||}|r"t|dkDr|j |d|j j |Scc}w)Nrr ziConversion to shapely line string requires at least two points, but points input contains only %d points.)rr7rr rfrr)pointsrr rr; points_tupless r!r r qs 6{a 8;>v;. IJ J8>>veeAha)vM>Q*=+F #f+/]1-.    & &} 55?sBczeZdZ d dZddZdZdZedZdZ edZ ddZ d Z ed Z d Zy )_ConcavePolygonRecovererc||_||_||_||_d|_d|_d|_d|_d|_d|_ d|_ d|_ y)NKdrOr riP) threshold_duplicate_pointsnoise_strength oversamplingmax_segment_differenceoversample_up_to_n_points_maxfit_n_changes_maxfit_n_iters_maxfit_max_dist_first_iterfit_max_dist_other_iters fit_n_candidates_before_sort_max$limit_coords_values_for_inter_searchdecimals)r:rrrrs r!r=z!_ConcavePolygonRecoverer.__init__so*D',(&<# .0*"% !()$()%14-5:1  r#ct|tsJtj|r!|jdk(r|j ddk(sJdt |dt|dk\sJd|j|}|jr|Stj|}|jd}|j|}|j||d}|j||d}|j!||j" }|j$!|j$dkDr|j'||}|j)||}|j+||d}|D cgc]} ||  } } |j| Scc} w) Nrr z5Expected exterior as list or (N,2) ndarray, got type rrOz=Cannot recover a concave polygon from less than three points.r?r)r)rrr3r4r5r6rr7rcrQiarandomRNG duplicate$_remove_consecutive_duplicate_points_fix_polygon_is_line_jitter_duplicate_points_generate_intersection_pointsrr_oversample_intersection_points_insert_intersection_points_fit_best_valid_polygon) r: new_exterior old_polygonrrrsssegment_add_pointsnew_exterior_internew_exterior_concave_idsidxnew_exterior_concaves r!rz%_ConcavePolygonRecoverer.recover_froms,- NN< (!!Q&""1%* * &)  *, < A% L K L% && &=   N||L1 $$Q'@@N 00s1vF 44\3q6J "?? 4==@2    (T->->-B!%!E!E0"2 "== ,.$(#?#? A$( 0H I/G s #/G  I ##-A#BB  Is& Fcg}|D]}|rotjjtj|tj|dz }||jk}|rb|j |t|j |t |dk\rftjjtj|dtj|dz }||jk}|r|ddn|}|S)Nr,rr)r.linalgnormr0rrfr7)r:rrr;rmis_sames r!rz=_ConcavePolygonRecoverer._remove_consecutive_duplicate_pointssEyy~~JJu% 6":(>>@$"A"AAMM%( e$ v;! 99>> 6!9% 6":(>>@Dd===G%,VAb\&F r#ct|dk\sJdt|fz|j}|j|r|j| |t|dfj t j }t||Dcgc]\}}|d|dz|d|dzf}}}|dz}|dkDs Jd|fz|j|r|Scc}}w) NrOzfCan only fix line-like polygons with an exterior containing at least 3 points. Got one with %d points.rrrr +Expected noise strength to be >0, got %.4f.)r7r_is_polygon_lineuniformrr.r0zip)r:rrrnoiser;noise_is r!rz-_ConcavePolygonRecoverer._fix_polygon_is_lines 8}! J 69._find_duplicates s$006J%&>?5q%)T-L-L$L"NOPq%)T-L-L$L"NOP",J&0 "AjL!J,#?@GGJ'1#-@ 3'?#@@J!'_ !*:!6IAv']F5f=F!&) "1Q3J8!+A!9&!A%f-$!ww#AY2Q6%ay6!94q89 : $"A"AA15Jv.9 "7"( r#rrFrrr rT)ranyrr7rr.r0r) r:rrrr convergedrrr is_duplicates ` r!rz1_ConcavePolygonRecoverer._jitter_duplicate_points s# J,,! O 9^zH_ConcavePolygonRecoverer._generate_intersection_points.. s!r#r)!r.rWabsrr0rr3warnrr7r4rrrallr7rrIfloat'imgaug.external.poly_point_isect_py2py3rr traceback print_excrrrXrfrrrhsortedr)"r:rone_point_per_intersectionr largest_valuetoo_large_values_r;rsegmentsr intersectionsexcsegments_add_pointsassociated_segments seg_inter_p0 seg_inter_p1diffsdistsseg_p0seg_p1 dist_p0p0 dist_p1p1 dist_p0p1 dist_p1p0diffmin_diffrsegment_add_points_sortedrrbotha_bs" r!rz6_ConcavePolygonRecoverer._generate_intersection_pointsi srvvbhhxrzz&JKL DEE E   GG+==.@ @ !&c(m 45 41B 45 5 >>( #H~H(D) 0"8n / 0):CJ!O:; E  Au3s5z?AB D E; x=A I$3x=) *HHU8A;q>2H=HHU8A;q>2H= HHU8QUc(m,C#DQ#GH%'HHU8QUc(m,C#DQ#GH%' *  " - 6;HEM",1X+?@+?ar+?@*7 &E&*':1'=&>#.A* l&.NFF " v8N/N OI " v8N/N OI " v8N/N OI " v8N/N OIy94i)6KLDLL&LLU1X-vay58/CD"'/66%=d999bii./C',33UE#J4GHGG./'/B +8B%'!012C$7$<=$< 2Gs\ S/SS *B9S",S' U U % U%; U* ' U0A U< UU UUc |j|jdkr|Stt|Dcgc]}g}}t|}t|D]b\}}t||D]J\}} | d|dz | d|dz f} |dk(rd|jz g} n|jd|jz g} | D]2} |d| | dzz|d| | dzzf} ||j | 4||j | | }|t||dz k(}|rt||dzt|z}|d|dz |d|dz f} ||j |d|j| dzz|d|j| dzzf||d}|t||z }||j kDsG|ccSe|Scc}w)Nrr r<r,)rrr7rrfr)r:rrrsegment_add_points_sorted_oversn_pointsrlastrp_inter direction oversamplerm point_overis_last_in_groupexterior_points r!rz8_ConcavePolygonRecoverer._oversample_intersection_points sA    $(9(9Q(>% %c"456+861B6 (+8x= *GAt'(:1(=> 7$QZ$q'171:Q3GH 6"%(9(9"9!:J#'"3"3S4;L;L5L!MJ&D"&q'D9Q<,?"?"&q'D9Q<,?"?"AJ3A6==jI'0299'B$%-?-B)Ca)G$G #&.q1uH .E%FN!/!2T!W!I3A6==a4#4#4y|#CCa4#4#4y|#CCE;1=bADC ? BCCd@@@::A?+F/.O+8s F=ct|t|k(sJdt|t|fzg}t|D]6\}}||}|j|||D]}|j|8|S)NzExpected one entry in 'segment_add_points' for every point in the exterior. Got %d (segment_add_points) and %d (exterior) entries instead.)r7rrf)rrrexterior_interprrrs r!rz4_ConcavePolygonRecoverer._insert_intersection_points s8}$6 77 K "%&8"93x=!I J K7"8,IAva[F  " "6 *-a0&&w/1- r#c 6 %t|dkryd}t|}|jrtjt|St j j|}t|j}tt|Dcgc] }||vs| }}d} d} d} | sg%t|D cic]\}} | | } }} tjtjtj|ttj|tj tj|ttj|g}tj"|tj$|j&ddftjfd}|dddf|dddfk}|ddddf|dddd f<||df||df<||df||df<tj(|dddf|dddfz t||dddfz |dddfz|ddd f<|j*}| dkDr |j,}||ddd f|k}|j.duxrt||j.kD}|r |j1||d|j.}|D]^}|d}|d}| |}|}||dzt|z}||}||}| dk(rd}n |||||}%j3|||d |f`tj4t%}t7|%fd  }|j8|d|j8}d}t;}|D]}%|d}%|d}||f|vst|D cgc] \}} | |k(r|c} }d} |dd}!|!j=| dz|t|!D"cgc]}"||" c}"}#|#jr0d }|!}|D cgc] } | |k7r|  }} | dz } | |j8k\r|cS|j?||f|j?||f|s| dkDrd } | dz } |j@duxr| |j@kD}$|$r |S| s|Scc}wcc} }wcc} }wcc}"wcc} w) NrcB|d|dz }|d|dz }t||dz||dzz |d|dzz|d|dzz }tj|dz|dzz}|dk(r0tj|d|dz dz|d|dz dzzS||z S)Nrr r)rr.rg)r; line_startline_endx_diffy_diffnumdens r!_compute_distance_point_to_linezY_ConcavePolygonRecoverer._fit_best_valid_polygon.._compute_distance_point_to_line# sa[:a=0Fa[:a=0FuQx&q/11+jm+,.6qk*Q-.GHC''&!)fai/0Caxww1X 1 -1Qx*Q-/!34559 r#rFrOr*r rrc$|d|dfS)NrrOr)rrs r!rzB_ConcavePolygonRecoverer._fit_best_valid_polygon.. sC!3Z_Q5G Hr#rT)!r7r%rQsmxrangescipyspatial ConvexHullrverticesrrr. transposerrHrepeatrr/r6minimumrrrshufflerfarangerrrinsertrr)&r:rrrrUhull points_keptr points_left iteration n_changesrrpoint_kept_idx_to_poscombosrmax_distcandidate_rowsdo_limitrowpoint_left_idxpoint_kept_idxin_points_kept_possegment_start_idxsegment_end_idx segment_start segment_end dist_eucl candidate_idsrQdone candidate_idxin_points_kept_idxpoints_kept_hypothesisrpoly_hypothesishas_reached_iters_maxrs& @r!rz0_ConcavePolygonRecoverer._fit_best_valid_polygon s] v;? v ==99S[) )}}''/4==) "'F "4M"4Q8Lq"4 M   J"2;;1G%I1GI 1 1G "%I \\HH[)3rxx /D+E HH[)3rxx /D+E #F^^6<<?A"6bhhGHF !Q$<&A,.D#AqsF^F1ac6N$T1WoF47O$T1WoF47O::q!t vad|+F fQTl*VAqD\9F1a4L 44H1}77#F1a4LH$<=N55TAP'$*O*OO $$^4!/d;;"=&!$Q!$Q%:>%J"$2!"-')S-==#? &'8 9 $_5 > !I ?~. {!LI!!#^SVYGI&(IIc*o6M"HJM%%1 -.Et/E/E F H5D!. !+M!:1!=!+M!:1!="N34?%[1*2)Ay$61*  *& .9^**11*1,&('./E/G/Es /E/G'HO&//#'&< +6'G*5,5*3~*E(1*5$'G"Q $(>(>>#..HHnn=>HHnn=>7"/> A  NI$$D05 4 44 "%YXcN,%Il*/G 'Gs$ RR< RR  R 8RN)r=r=rr=)r)Tr)rArBrCr=rrr classmethodrrrrrrrrr#r!rrs~GK;?,\1Cf&"  >B ! !BF/0u)n-/^  ir#rc(eZdZdZdZeddZy)rz Class that represents several polygons. Parameters ---------- geoms : list of imgaug.augmentables.polys.Polygon List of the polygons. c t|dk(sWt|Dcgc]}t|tc}s0Jddj |Dcgc] }t |c}z||_ycc}wcc}w)z#Create a new MultiPolygon instance.rz>Expected 'geoms' to a list of Polygon instances. Got types %s.r6N)r7rrr%r7rr)r:rels r!r=zMultiPolygon.__init__ sy J!Oe =-Fs A.A3 Nc `ddl}t||jjr8t|jDcgc]}t j ||c}St||jj r!tt j ||gSt||jjjrt|jDcgc]"}t||jj $c}sCJddj|jDcgc]}tt|c}zt|jDcgc]}t j ||c}Stdt|dcc}wcc}wcc}wcc}w)aWCreate a MultiPolygon from a shapely object. This also creates all necessary ``Polygon`` s contained in this ``MultiPolygon``. Parameters ---------- geometry : shapely.geometry.MultiPolygon or shapely.geometry.Polygon or shapely.geometry.collection.GeometryCollection The object to convert to a MultiPolygon. label : None or str, optional A label assigned to all Polygons within the MultiPolygon. Returns ------- imgaug.augmentables.polys.MultiPolygon The derived MultiPolygon. rNrzPExpected the geometry collection to only contain shapely polygons. Got types %s.r6zUnknown datatype 'zy'. Expected shapely.geometry.Polygon or shapely.geometry.MultiPolygon or shapely.geometry.collections.GeometryCollection.)rrrrrr%r collectionrrr7rrr)rr9rrUvs r!rzMultiPolygon.from_shapely s, h 0 0 = = >>>!#"$$T$7!!#$ $ h 0 0 8 8 9!5!5he!5!L MN N h&&11DD F>>#"4!1!1!9!9:!#$ L. "HA3tAw<"HIK  L$ >>!#"$$T$7!!#$ $ X !" ")!## #I!#sF'F!$F& F+r)rArBrCrDr=rGrrr#r!rr s 0"0"r#rr>)%rD __future__rrrrrnumpyr.scipy.spatial.distancer" six.movesmovesr  skimage.drawrskimage.measurerwrr3rrbaser utilsr r r rrr"objectr%rIr rrrr#r!rWs$@@!++/lC8fC8N*Z ,lZ ,z;@7860vHD"6D"r#