o e@@sdZddlmZmZmZddlZddlZddlZddl Z ddl m Z ddlZddlZddlmZddlmZddlmZdd lmZmZmZmZmZd d ZGd d d eZ GdddeZ!  dddZ"GdddeZ#GdddeZ$dS)zClasses dealing with polygons.)print_functiondivisionabsolute_importN)imgaug)random) IAugmentable)normalize_shapeinterpolate_points_remove_out_of_image_fraction_project_coords__normalize_shift_argsc Csvd}t|tsd}|g}|g}t|D]\}}t|jD]\}}||j||j||} | j|_qq|s9|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) ZpsoisZ psois_origZ recoverer random_stateZinput_was_listiZpsoijpolygonZpoly_recrID:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/polys.pyrecover_psois_s    rc@seZdZdZddddZeddZeddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZddZddZdeddZddZd d!Zd"d#Zd$d%Zdfd'd(Zejd)d*d+d,d-Zd.d/Zdgd1d2Zdhd3d4Z      did8d9Zd:d;Z < &djd=d>Zd?d@Z dAdBZ!dCdDZ"dEdFZ#dkdGdHZ$dIdJZ%dKdLZ&dldMdNZ'e(dddOdPZ) < QdmdRdSZ* < QdmdTdUZ+dmdVdWZ,dndXdYZ-dndZd[Z.d\d]Z/d^d_Z0d`daZ1dbdcZ2dS)oPolygonaClass 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. NcCs ddlm}t|tr6|stjdtjd|_nEt|d|r*tdd|D|_n2tdd|D|_n&t |sBJd|f|j d krN|j d d ksVJd |j ft||_t |jd kovt |jdd d f|jd d d f}|r|jd d |_||_d S)zCreate a new Polygon instance.rKeypointrrdtypecSsg|]}|j|jgqSrxy.0pointrrr rsz$Polygon.__init__..cSsg|] }|d|dgqSrrrr%rrrr(wszOExpected exterior to be a list of tuples (x, y) or an (N, 2) array, got type %srrz\Expected exterior to be a list of tuples (x, y) or an (N, 2) array, got an array of shape %sN)imgaug.augmentables.kpsrrrnpzerosfloat32ria is_np_arrayndimshapelenZallcloselabel)selfrr4r is_closedrrr__init__es:      & zPolygon.__init__cC|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``. rr5rrrcoordss zPolygon.coordscC|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. Nrr9r:rrrxx z Polygon.xxcCr<)zGet the y-coordinates of all points on the exterior. Returns ------- (N,2) ndarray ``float32`` y-coordinates array of all points on the exterior. Nrr9r:rrryyr>z Polygon.yycCtt|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,int32roundr=r:rrrxx_intzPolygon.xx_intcCr@)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,rArBr?r:rrryy_intrDzPolygon.yy_intcCst|jdkr dS|jS)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)r3rto_shapely_polygonis_validr:rrrrHs zPolygon.is_validcCs t|jdkr dS|}|jS)z{Compute the area of the polygon. Returns ------- number Area of the polygon. rFg)r3rrGarea)r5polyrrrrIs z Polygon.areacC|j}t|t|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. )r?maxmin)r5r?rrrheight zPolygon.heightcCrK)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. )r=rLrM)r5r=rrrwidth rOz Polygon.widthcCst|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. )r r;rr5Z from_shapeZto_shaperrrproject_szPolygon.project_cC|||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. )deepcopyrRrQrrrproject9szPolygon.projectFc Cstt|jdks Jdg}|jD]\}}||d||d}||qt|}t|}|r8|||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)r3rappendr,sqrtargmin) r5r#r$return_distance distancesx2y2dist closest_idxrrrfind_closest_point_indexTs    z Polygon.find_closest_point_indexcCs4||}t|dkr|jS|jtdd|DS)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``. rcSg|]}|jqSr)rIr&rJrrrr(z5Polygon.compute_out_of_image_area..)clip_out_of_imager3rIsum)r5imageZ polys_clippedrrrcompute_out_of_image_area~s  z!Polygon.compute_out_of_image_areacCs*|j}|dkr||S|||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)rIto_line_stringcompute_out_of_image_fractionrf)r5rerIrrrrhsz%Polygon.compute_out_of_image_fractioncCs|j|ddd 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_imager5rerrris_fully_within_imagezPolygon.is_fully_within_imagecCs|j|ddd 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. TFrirlrnrrris_partly_within_imagerpzPolygon.is_partly_within_imageTcCs^t|jdkr td|}||rdS|j|dddr |S||}t|dkr-|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.FTri)r3r Exceptionrgrormrc)r5rerjrklspolysrrrrms   zPolygon.is_out_of_imagezPolygon.clip_out_of_image()z3clip_out_of_image() has the exactly same interface.)Zalt_funccommentcCs ||S)z>Cut off all parts of the polygon that are outside of an image.)rcrnrrrcut_out_of_images zPolygon.cut_out_of_imagecCsddl}t|jdkr gSt|jdvr9|jdd}||}t|dks'Jt|dkr/gS|j|djdgSt|rE|j ddn|dd\}}| }|j d |df||fd|fg}| |} |j j|j j|j jj|j jf} t| |j j r|j | g} n6t| |j jrn.t| | r|j g} n"t| |j jr| jsJgSt| ||jtd t| |||jfg} | jD] } | t j| |jd qg} | D]3}d}d}|jD]\}}|j||d d \}}|dus||kr|}|}q|dur||}| |q| 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)rrFclosedrr9rrrz}Got an unexpected result of type %s from Shapely for image (%d, %d) and polygon %s. This is an internal error. Please report.r4Tr#r$rY) shapely.geometryr3rrgrcrTr;r/r0r2rGgeometryr intersection LineStringZMultiLineStringr'ZPointZ MultiPointr MultiPolygonGeometryCollectionZis_emptyprintrrtypegeomsrV from_shapelyr4r_change_first_point_by_index)r5reshapelyrsZ ls_clippedhwZ poly_shapelyZ poly_imageZmultipoly_inter_shapelyZ ignore_typesrZpoly_inter_shapelyZpolygons_reorderedrZbest_idxZ best_distr#r$ point_idxr]Zpolygon_reorderedrrrrc#s   (           zPolygon.clip_out_of_imagercCs8|jdddf|7<|jdddf|7<|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. Nrrr9)r5r#r$rrrshift_szPolygon.shift_cC*t||||||d\}}|j||dS)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. toprightbottomleftr"rrTrr5r#r$rrrrrrrshift)  z Polygon.shiftrr?rcCs:dd}dd}|d||d||d| ||t|}||t|d}||t|d}|||d}|||}|| |} || | } || | d } |jd krgt|scJd t|f|g}n|jd kryt|ry|g|jd }|d krd}n|dkrd}| r||rtdt||jf|j }| tj }t j j|j|j|jd\}}t|dkr|dkrt ||||f<n|dkrnd||||ddf|t ||||f<|jdd}|jdd}|j|||| | d}|j||| | | d}|jtjkrtt|dd |}|S| |}|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. cSs |dusJd|t|fdS)Nz*Expected '%s' to not be None, got type %s.)r)Zarg_name arg_valuerrr_assert_not_none2s z/Polygon.draw_on_image.._assert_not_nonecSs|dur|S|SNr)vardefaultrrr _default_to7sz*Polygon.draw_on_image.._default_tocoloralphasizeg?rFrzMGot a 2D image. Expected then 'color_face' to be a single number, but got %s.r*{Gz?rgGz?rz.Cannot draw polygon %s on image with shape %s.r2NFrwT)rrrraise_if_out_of_imager)r,arrayr1r/Zis_single_numberstrr2rmrrr!astyper.skimagedrawrrErCr3rgZdraw_lines_on_imageZdraw_points_on_imagerZuint8ZcliprB)r5rer color_face color_lines color_pointsr alpha_face alpha_lines alpha_pointsr size_lines size_pointsrrrZ input_dtyperesultrrccZls_openZ ls_closedrrr draw_on_imagesv[                zPolygon.draw_on_imagec Cs |jdvs Jd|jft|jdkrtd|}||}|j|dddr+|S|j}|j }|t |}|t |}t |}t |} t jj|||| fd\} } t j|| ft jd } d| | | f<|jd krt | d d d d t jfd d |jdf} || S) 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. )rrFz0Expected image of shape (H,W,[C]), got shape %s.rzOPolygon must be made up of at least 3 points to extract its area from an image.TFrirr rFNr)r1r2r3rrrto_bounding_boxextract_from_imagermrCrEr,rMrLrrrr-booltileZnewaxis) r5reZbbZbb_arear=r?Zxx_maskZyy_maskZ height_maskZ width_maskZrr_faceZcc_facemaskrrrrs2        ,zPolygon.extract_from_image-C6?cCs|t|jdkr td|j||dd\}}|dur9||kr9|s#|S|j|ddf}td|d|d|f||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.Tr{NzFClosest found point (%.9f, %.9f) exceeds max_distance of %.9f exceededr)r3rrrr_rTr)r5r#r$ max_distanceZraise_if_too_far_awayr^Z closest_distZ closest_pointrrrchange_first_point_by_coordss "  z$Polygon.change_first_point_by_coordscCsd|kr t|jksnJdt|j|f|dkr!|Stj|j|dddf|jd|ddffdd}|j|dS)a Reorder exterior points so that the point with given index is first. This method takes a given index and reorders all exterior corner points so that the point with that index becomes the first one in the array. An ``AssertionError`` will be raised if the index does not match any exterior point's index or the exterior does not contain any points. Parameters ---------- point_idx : int Index of the desired starting point. Returns ------- imgaug.augmentables.polys.Polygon Copy of this polygon with the new point order. rzWExpected index of new first point to be in the discrete interval [0..%d). Got index %d.NZaxisr9)r3rrTr, concatenate)r5rrrrrrs * z#Polygon.change_first_point_by_indexcCs@t|jdkr |S|jdd}||}|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. rTrwNr*)r3rrg subdivider;)r5points_per_edgersZls_subZexterior_subdividedrrr subdivide_s  zPolygon.subdivide_cC||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. rTrr5rrrrr/szPolygon.subdividecCs ddl}|jdd|jDS)zConvert this polygon to a ``Shapely`` ``Polygon``. Returns ------- shapely.geometry.Polygon The ``Shapely`` ``Polygon`` matching this polygon's exterior. rNcSg|] }|d|dfqSr)rr%rrrr(Pz.Polygon.to_shapely_polygon..)r|r}rr)r5rrrrrGCs zPolygon.to_shapely_polygoncCst|j||dS)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. )rx interpolate)&_convert_points_to_shapely_line_stringr)r5rxrrrrto_shapely_line_stringRszPolygon.to_shapely_line_stringcCs<ddlm}|j}|j}|t|t|t|t||jdS)zConvert this polygon to a bounding box containing the polygon. Returns ------- imgaug.augmentables.bbs.BoundingBox Bounding box that tightly encapsulates the polygon. r) BoundingBox)x1r[y1r\r4)Zimgaug.augmentables.bbsrr=r?rMrLr4)r5rr=r?rrrrhs  zPolygon.to_bounding_boxcs ddlmfdd|jDS)zConvert this polygon's exterior to ``Keypoint`` instances. Returns ------- list of imgaug.augmentables.kps.Keypoint Exterior vertices as :class:`~imgaug.augmentables.kps.Keypoint` instances. rrcs g|] }|d|ddqS)rrr"rr%rrrr(s z(Polygon.to_keypoints..)r+rrr:rrr to_keypointszs zPolygon.to_keypointscCs^ddlm}|rt|jdkr||j|jdS|tj|j|jddddfgdd|jdS)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)rrrzNr)Zimgaug.augmentables.linesrr3rr4r,r)r5rxrrrrrgs $zPolygon.to_line_stringcCsvddl}t||jjsJdt|f|jdup t|jjdk}|r)tg|dSt dd|jjD}t||dS)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``. rNzDExpected the input to be a shapely.geometry.Polgon instance. Got %s.rzcSsg|]\}}||gqSrr)r&r#r$rrrr(s z(Polygon.from_shapely..) r|rr}rrrr3r;r,r.)Zpolygon_shapelyr4rZhas_no_exteriorrrrrrs    zPolygon.from_shapelycCs|j|||dS)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_equalsr5otherrrrrrcoords_almost_equalsszPolygon.coords_almost_equalscCsjt|tr tt|}nt|rt|}nt|ts%Jdt|f|jddj |jdd||dS)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). zbExpected 'other' to be a list of coordinates, a coordinate array or a single Polygon. Got type %s.Trwr) rrrr,r.r/r0rrgrrrrrrs ,     zPolygon.exterior_almost_equalscCs |j|jkrdS|j|||dS)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)r4rrrrr almost_equals s zPolygon.almost_equalscCs|j||dS)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. rr4rTr5rr4rrrcopyAsz Polygon.copycCs2t|dur t|jn||dur|jdS|dS)aCreate a deep copy of this object. Parameters ---------- exterior : list of Keypoint or list of tuple or (N,2) ndarray, optional List of points defining the polygon. See `imgaug.augmentables.polys.Polygon.__init__` for details. label : None or str If not ``None``, the ``label`` of the copied object will be set to this value. Returns ------- imgaug.augmentables.polys.Polygon Deep copy. Nr)rr,rrr4rrrrrTVs  zPolygon.deepcopycC |j|S)zGet the coordinate(s) with given indices. Added in 0.4.0. Returns ------- ndarray xy-coordinate(s) as ``ndarray``. r9r5indicesrrr __getitem__m zPolygon.__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. )iterrr:rrr__iter__zrzPolygon.__iter__cC|Sr__str__r:rrr__repr__zPolygon.__repr__cCs,ddd|jD}d|t|j|jfS)N, cSs g|] }d|d|dfqS)z(x=%.3f, y=%.3f)rrrr%rrrr(sz#Polygon.__str__..z#Polygon([%s] (%d points), label=%s))joinrr3r4)r5Z points_strrrrrs zPolygon.__str__r)FTFryrrNNNN rNNNrNNNrNNF)rTFr)T)rrNN)3__name__ __module__ __qualname____doc__r7propertyr;r=r?rCrErHrIrNrPrRrUr_rfrhrorqrmr/ deprecatedrvrcrrrrrrrrrGrrrrg staticmethodrrrrrrTrrrrrrrrrMs +          *" 8  i 1 54 3!   '  ; !   rc@seZdZdZddZeddZejddZeddZd d Z d d Z      d@ddZ dAddZ dAddZ ddZddZddZdd ZdBd"d#ZdCd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZdDd2d3ZdDd4d5Zd6d7Zd8d9Zd:d;Zdd?Z dS)EPolygonsOnImageaContainer 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) cCs||_t||_dSr)rr r2r5rr2rrrr7szPolygonsOnImage.__init__cCr8)zGet the polygons in this container. Added in 0.4.0. Returns ------- list of Polygon Polygons within this container. rr:rrritemss zPolygonsOnImage.itemscCs ||_dS)zSet the polygons in this container. Added in 0.4.0. Parameters ---------- value : list of Polygon Polygons within this container. Nr)r5valuerrrrs cCst|jdkS)zEstimate whether this object contains zero polygons. Returns ------- bool ``True`` if this object contains zero polygons. r)r3rr:rrremptys zPolygonsOnImage.emptycCs^t|}|dd|jddkr||_|St|jD]\}}||j||j|<q||_|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 r2rrrRr)r5reZon_shaperitemrrron_szPolygonsOnImage.on_cCr)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. )rTrrnrrronzPolygonsOnImage.onrNrrFcCs4|jD]}|j||||||||| | | | | d }q|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)r5rerrrrrrrrrrrrrJrrrr s" \zPolygonsOnImage.draw_on_imageTcsfddjD_S)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. cs"g|] }|jjds|qS)ri)rmr2rarjrkr5rrr(s z8PolygonsOnImage.remove_out_of_image_..rr5rjrkrrrremove_out_of_image_wsz$PolygonsOnImage.remove_out_of_image_cCrS)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. )rTrrrrrremove_out_of_imagesz#PolygonsOnImage.remove_out_of_imagecCs t||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 r5fractionrrrremove_out_of_image_fraction_s z-PolygonsOnImage.remove_out_of_image_fraction_cCr)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. )rrrrrrremove_out_of_image_fractionsz,PolygonsOnImage.remove_out_of_image_fractioncsfddjD_S)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. cs"g|] }|jD]}|q qSr)rcr2)r&rJZ poly_clippedr:rrr(s z6PolygonsOnImage.clip_out_of_image_..rr:rr:rclip_out_of_image_s z"PolygonsOnImage.clip_out_of_image_cCs |S)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. )rrr:rrrrcs z!PolygonsOnImage.clip_out_of_imagercCs,t|jD]\}}|j||d|j|<q|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)r5r#r$rrJrrrrszPolygonsOnImage.shift_cCr)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. rr"rrrrrr+rzPolygonsOnImage.shiftcCs(t|jD] \}}|||j|<q|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)r5rrrJrrrrXszPolygonsOnImage.subdivide_cCrrrrrrrrlrzPolygonsOnImage.subdividecCs,|jr tjdtjdStdd|jDS)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. rr cSr`rr9rarrrr(rbz/PolygonsOnImage.to_xy_array..)rr,r-r.rrr:rrr to_xy_array~s zPolygonsOnImage.to_xy_arraycCstj|tjd}|jddks#|jdkr|jddks#Jd|jfd}|jD]4}t|j}||t|ksIJdt|tdd|jDf|||||jd d d f<||7}q(|t|ksmJd t||f|S) 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*z7Expected input array to have shape (N,2), got shape %s.zpReceived fewer points than there are corner points in the exteriors of all polygons. Got %d points, expected %d.cSsg|]}t|jqSr)r3r)r&prrrr(z7PolygonsOnImage.fill_from_xy_array_..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,rr.r2r1rr3rrd)r5ZxycounterrJZ nb_pointsrrrfill_from_xy_array_s0(   z#PolygonsOnImage.fill_from_xy_array_cCsJddlm}|jr|g|jdStjdd|jDdd}|j||jdS)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)KeypointsOnImagercSr`rr9rarrrr(rbz9PolygonsOnImage.to_keypoints_on_image..rr)rrr2r,rrZ from_xy_array)r5r exteriorsrrrto_keypoints_on_images z%PolygonsOnImage.to_keypoints_on_imagec Cs|j}dd|D}tdd|D}t|j|ks%Jd|t|jf|}d}|D]!}|j}|||t|ddf|ddddf<|t|7}q-|j|_|S)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. cSr`rr9rarrrr(rbzAPolygonsOnImage.invert_to_keypoints_on_image_..cSg|]}t|qSrr3)r&rrrrr(z Expected %d coordinates, got %d.rN)rrdr3Z keypointsr rr2) r5ZkpsoirtrZ nb_points_expZxy_arrr rJrrrrinvert_to_keypoints_on_image_s  ,z-PolygonsOnImage.invert_to_keypoints_on_image_cCs2|dur |jdd}|durt|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. Nrtupler2rrrrrrs   zPolygonsOnImage.copycCs4|dur dd|jD}|durt|j}t||S)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. NcSsg|]}|qSrrrarrrr(9rz,PolygonsOnImage.deepcopy..rrrrrrT s   zPolygonsOnImage.deepcopycCr)zGet the polygon(s) with given indices. Added in 0.4.0. Returns ------- list of imgaug.augmentables.polys.Polygon Polygon(s) with given indices. rrrrrr@rzPolygonsOnImage.__getitem__cCr)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. )rrr:rrrrMs zPolygonsOnImage.__iter__cCr)zGet the number of items in this instance. Added in 0.4.0. Returns ------- int Number of items in this instance. )r3rr:rrr__len__\rzPolygonsOnImage.__len__cCrrrr:rrrrirzPolygonsOnImage.__repr__cCsdt|j|jfS)NzPolygonsOnImage(%s, shape=%s))rrr2r:rrrrls zPolygonsOnImage.__str__rrryrr)!rrrrr7rrsetterrrrrrrrrrrcrrrrr r rrrrTrrrrrrrrrrsN     n "  -8 &    rFcCspddl}t|dkrtdt|fdd|D}|dkr#t||}|r2t|dkr2||d|j|S)NrrziConversion to shapely line string requires at least two points, but points input contains only %d points.cSrr)rr%rrrr({rz:_convert_points_to_shapely_line_string..)r|r3rrr rVr}r)pointsrxrrZ points_tuplesrrrrqs   rc@s~eZdZ  dddZdddZdd Zd d Zed d ZddZ eddZ  dddZ ddZ eddZ ddZdS) _ConcavePolygonRecovererrrcCsL||_||_||_||_d|_d|_d|_d|_d|_d|_ d|_ d|_ dS)NKdrFrriP) 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)r5rr r!r"rrrr7s z!_ConcavePolygonRecoverer.__init__rc st|tst|r|jdkr|jddksJdt|ft|dks)Jd|j|d}|j r4|St |}| d}| |}|||d}|||d}|j||jd}|jdurk|jdkrk|||}|||||d}fd d |D}|j|dS) Nrrz8Expected exterior as list or (N,2) ndarray, got type %s.rFz=Cannot recover a concave polygon from less than three points.r9r)r*cg|]}|qSrrr&idxZnew_exterior_interrrr(z9_ConcavePolygonRecoverer.recover_from..)rrr/r0r1r2rr3rTrHiarandomZRNGZ duplicate$_remove_consecutive_duplicate_points_fix_polygon_is_line_jitter_duplicate_points_generate_intersection_pointsr*r!_oversample_intersection_points_insert_intersection_points_fit_best_valid_polygon) r5Z new_exteriorZ old_polygonrrZrsssegment_add_pointsZnew_exterior_concave_idsZnew_exterior_concaverr.rrsJ        z%_ConcavePolygonRecoverer.recover_fromcCsg}|D]&}|r%tjt|t|d}||jk}|s$||q||qt|dkrRtjt|dt|d}||jk}|rP|ddn|}|S)Nr*rr)r,linalgnormr.rrVr3)r5rrr'r]Zis_samerrrr1s$     z=_ConcavePolygonRecoverer._remove_consecutive_duplicate_pointscCst|dksJdt|f|j}||rF|j| |t|dfdtj}ddt||D}|d}|dksAJd |f||s|S) NrFzfCan only fix line-like polygons with an exterior containing at least 3 points. Got one with %d points.rrcSs0g|]\}}|d|d|d|dfqSr)r)r&r'Znoise_irrrr( s$zA_ConcavePolygonRecoverer._fix_polygon_is_line.. r+Expected noise strength to be >0, got %.4f.)r3r _is_polygon_lineuniformrr,r.zip)r5rrr noiserrrr2s.    z-_ConcavePolygonRecoverer._fix_polygon_is_linecCsltddg}|d}t}|ddD]}t|t|}t||}|t|dqt|dkS)Nrri)r,r.setr/Zangle_between_vectorsaddintr3)clsrZvec_downpoint1ZanglesZpoint2ZvecZanglerrrr> s  z)_ConcavePolygonRecoverer._is_polygon_linec sfdd}j}|dksJd|f|dd}d}|sg||}t|rc|jj jt|dfdtj}t|D] \}} | r]||d||d||d||df||<q=|d 9}nd }|r|S) Nc sDtt}t|D]7\}}tt|ddj}tt|ddj}dD]}dD]}|||||f|q/q+q dgt |}|D]U} || } t| D]J\}} | |} || } || reqTt |dt | D]/} | | }||}||r}qnt | d|dd| d|dd}|jkrd||<qnqTqJ|S)Nrg?r)r*rrFrT) collections defaultdictrrrDr,rBrrVr3rangerW)Zexterior_with_duplicatesZ points_maprr'r#r$Z direction0Z direction1 duplicateskey candidatesZp0_idxpoint0rZp1_idxrFr]r:rr_find_duplicates sJ    zK_ConcavePolygonRecoverer._jitter_duplicate_points.._find_duplicatesrr=Frr;rr<T)r anyr?r3rr,r.r) r5rrrNr  convergedrJrArZ is_duplicaterr:rr3 s6 %    z1_ConcavePolygonRecoverer._jitter_duplicate_pointscCst|dksJdt|ftj|tjd}tjt|dftjd}||ddddf<|ddddf|ddddf<|dddf|dddf<tjj|ddddf|ddddfdd }t|S) NrFzLNeed at least 3 points on the exterior to compute the circumference. Got %d.r rrrrr*r)r3r,rr.r-r9r:rd)rErZ points_matrixrZrrr_calculate_circumference[ s$( z1_ConcavePolygonRecoverer._calculate_circumferenceTrc stttjtjd}||jk}|r)td|jfddtt DSt r2t t t s@Jdt ftddDsWJddd dDt d kr_gSfd dtt D}d d lm}z||}Wn+ty} ztd t| ftddtt DWYd} ~ Sd} ~ wwddtt |D} |D]\} } |r| d g} | D]\} }g}g}|D]U\}}tj|t| }tj|t|}tj|t|}tj|t| }t||||}|||tj|d | d |d| dfqt|}||jkr=tt|}| || ||fqtdqqg}tt | D]8}dd| |D}dd| |D}t |dkrn||qLtt||ddd}|dd|DqL|S)Nr awEncountered during polygon repair a polygon with extremely large coordinate values beyond %d. Will skip intersection point computation for that polygon. This avoids exceptions and is -- due to the extreme distortion -- likely pointless anyways (i.e. the polygon is already broken beyond repair). Try using weaker augmentation parameters to avoid such large coordinate values.cSg|]}gqSrrr&_rrrr({ zJ_ConcavePolygonRecoverer._generate_intersection_points..z;Expected 'exterior' to be a list or a ndarray. Got type %s.cSsg|]}t|dkqS)rrr%rrrr( zFExpected 'exterior' to contain (x,y) coordinate pairs. Got lengths %s.rcSg|]}tt|qSr)rr3r%rrrr( rVrc sg|]>}tt|dtt|dftt|dtdtt|dtdffqSr))r,rBfloatr3r&rr*rrrr( s )isect_segments_include_segmentszhEncountered exception %s during polygon repair in segment intersection computation. Will skip that step.cSrRrrrSrrrr( rUcSrRrrrSrrrr( rUrz^Couldn't find fitting segment in _generate_intersection_points(). Ignoring intersection point.cSg|]}|dqSrrr&trrrr( rcSr\)rrr^rrrr( rrcSs|dS)Nrr)r_rrr szH_ConcavePolygonRecoverer._generate_intersection_points..rKcSsg|]\}}|qSrr)r&a_brrrr( r)r,rLabsrr.r)r/warnrIr3r0rrrallrZ'imgaug.external.poly_point_isect_py2py3r[rrr traceback print_excr9r:rMrVr"rDrXsortedr@)r5rZone_point_per_intersectionr*Z largest_valueZtoo_large_valuessegmentsr[Z intersectionsexcZsegments_add_pointsr'Zassociated_segmentsZ seg_inter_p0Z seg_inter_p1ZdiffsdistsZseg_p0Zseg_p1Z dist_p0p0Z dist_p1p1Z dist_p0p1Z dist_p1p0diffZmin_diffr-Zsegment_add_points_sortedrZbothrrZrr4i s       "        z6_ConcavePolygonRecoverer._generate_intersection_pointsc Cs|jdus |jdkr |Sddtt|D}t|}t|D]\}}t||D]\}}|d|d|d|df} |dkrHd|jg} n|jd|jg} | D]} |d| | d|d| | df} ||| qR||||}|t||dk} | r||dt|}|d|d|d|df} |||d|j| d|d|j| df||d}|t||7}||jkr|Sq)q|S)NrcSrRrrrSrrrr( szL_ConcavePolygonRecoverer._oversample_intersection_points..rrr*)r!rIr3rrVr#)r5rr8Zsegment_add_points_sorted_oversZn_pointsrlastrp_inter directionZ oversampler]Z point_overZis_last_in_groupZexterior_pointrrrr5 sJ     "z8_ConcavePolygonRecoverer._oversample_intersection_pointscCslt|t|ksJdt|t|fg}t|D]\}}||}||||D]}||q+q|S)NzExpected one entry in 'segment_add_points' for every point in the exterior. Got %d (segment_add_points) and %d (exterior) entries instead.)r3rrV)rErr8Zexterior_interprrMrorrrr6 s   z4_ConcavePolygonRecoverer._insert_intersection_pointsc stdkrdSdd}t}|jrttStj}t|j fddt tD}d}d}d} | sgdd t D} t t t |tt t t tt |g} t j| t j| jdd ft jd fd d } | dddf| ddd fk} | ddddf| ddddf<| | d f| | df<| | df| | d f<t | ddd f| dddft| ddd f| dddf| dddf<|j} |dkr|j} | | dddf| k}|jduot||jk}|r |||d|j}|D]>}|d|d | }}|d t}|}|}|dkr8d}n|||}|d|fqt t}t|fddd}|jdurk|d|j}d}t}|D]i}|d|d f|vrڇfddt Dd}dd}||d tfdd|D}|jrd}|fdd|D}|d 7}||jkr̈S|f|fqr|s|dkrd} |d 7}|j duo||j k}|r S| r;S)NrcSs|d|d}|d|d}t||d||d|d|d|d|d}t|d|d}|dkrTt|d|dd|d|ddS||S)Nrrr)rdr,rW)r' line_startline_endZx_diffZy_diffnumZdenrrr_compute_distance_point_to_line# s"zY_ConcavePolygonRecoverer._fit_best_valid_polygon.._compute_distance_point_to_linecsg|]}|vr|qSrrrY) points_keptrrr(7 rVzD_ConcavePolygonRecoverer._fit_best_valid_polygon..rFcSsi|]\}}||qSrrr&rrrrr M s zD_ConcavePolygonRecoverer._fit_best_valid_polygon..rFr rrrcs|d|dfS)NrrFr)r-)rLrrr` rVzB_ConcavePolygonRecoverer._fit_best_valid_polygon..racsg|] \}}|kr|qSrrrv)point_kept_idxrrr( s cr+rrr,)rrrr( r/Tcsg|]}|kr|qSrr)r&r)point_left_idxrrr( s )!r3rrHsmxrangescipyZspatialZ ConvexHullrZverticesrIrr,Z transposerrArepeatrr-r2minimumr'r&r(shufflerVZarangerir$rBinsertrCr%)r5rrrtrJZhullZ points_left iterationZ n_changesrPZpoint_kept_idx_to_posZcombosrZmax_distZcandidate_rowsZdo_limitrowZin_points_kept_posZsegment_start_idxZsegment_end_idxZ segment_startZ segment_endZ dist_euclZ candidate_idsrHdoneZ candidate_idxZin_points_kept_idxZpoints_kept_hypothesisZpoly_hypothesisZhas_reached_iters_maxr)rLrxryrrurr7 s    $&                 z0_ConcavePolygonRecoverer._fit_best_valid_polygonN)rrrrr])Tr)rrrr7rr1r2 classmethodr>r3rQr4r5r6r7rrrrrs& .3  A  w/  rc@s&eZdZdZddZedddZdS)rz Class that represents several polygons. Parameters ---------- geoms : list of imgaug.augmentables.polys.Polygon List of the polygons. cCsDt|dkstdd|DsJdddd|D||_dS)z#Create a new MultiPolygon instance.rcSsg|]}t|tqSr)rrr&elrrrr( r z)MultiPolygon.__init__..z>Expected 'geoms' to a list of Polygon instances. Got types %s.rcSrr)rrrrrr( rN)r3rfrr)r5rrrrr7 s  zMultiPolygon.__init__Ncsddlt|jjrtfdd|jDSt|jjr(ttj|dgSt|jjjrWt fdd|jDsKJdd d d|jDtfd d|jDSt d t |f) 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. rNcg|] }tj|dqSrzrrrarzrrr(  z-MultiPolygon.from_shapely..rzcsg|] }t|jjqSr)rr}rra)rrrr( rzPExpected the geometry collection to only contain shapely polygons. Got types %s.rcSrWr)rr)r&vrrrr( rVcrrrrarzrrr( rzUnknown datatype '%s'. Expected shapely.geometry.Polygon or shapely.geometry.MultiPolygon or shapely.geometry.collections.GeometryCollection.) r|rr}rrrrZ collectionrrfrrrr)r}r4r)r4rrr s4    zMultiPolygon.from_shapelyr)rrrrr7rrrrrrr s   rr)%r __future__rrrrgrGnumpyr,Zscipy.spatial.distancer|Z six.movesmovesrzZ skimage.drawrZskimage.measurerrr/rr0baser utilsr r r r rrobjectrrrrrrrrrsP    6Qb H