\;j:/dZddlmZmZmZddlZddlZddl m Z ddl Z dZdZdZdZdZd Zdd Zdd Zd Zd ZdZddZy)z.Utility functions used in augmentable modules.)print_functionabsolute_importdivisionNctj|rtj|Sg}|D][}tj|r%|j tj|=|j |j ]|SN)ia is_np_arraynpcopyappenddeepcopy) augmentablesresult augmentables bG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\imgaug/augmentables/utils.pycopy_augmentablesr si ~~l#ww|$$ F# >>+ & MM"''+. / MM+..0 1 $ Mc|ytj|stj|r|St|tr|Dcgc] }t |c}St|t r"t |Dcgc] }t |c}Stj|rtj|St|dr|jStj|Scc}wcc}w)Nr ) ris_single_number is_string isinstancelist deepcopy_fasttupler r r hasattrr copylib)objels rrrs { 32<<#4 #t,/0Cb b!C00#u#6#BmB'#677 ~~cwws|sJ||~   C  16s C&2C+ct|tr|Stj|sJdt |d|j S)aNormalize a shape ``tuple`` or ``array`` to a shape ``tuple``. Parameters ---------- shape : tuple of int or ndarray The input to normalize. May optionally be an array. Returns ------- tuple of int Shape ``tuple``. z%Expected tuple of ints or array, got .)rrrr typeshape)r"s rnormalize_shaper#)sB% >>% E6:5kCE ;;rct|}t|}|dd|ddk(r|S|dd\}}|dd\}}t||||fDcgc]}|dkD c}}|sJd|d|d|} tj|r|jj dk7r2t j|jt j} | dddf|z |z| dddf<| dddf|z |z| dddf<| Scc}w) aProject coordinates from one image shape to another in-place. This performs a relative projection, e.g. a point at ``60%`` of the old image width will be at ``60%`` of the new image width after projection. Added in 0.4.0. Parameters ---------- coords : ndarray or list of tuple of number Coordinates to project. Either an ``(N,2)`` numpy array or a ``list`` containing ``(x,y)`` coordinate ``tuple`` s. from_shape : tuple of int or ndarray Old image shape. to_shape : tuple of int or ndarray New image shape. Returns ------- ndarray Projected coordinates as ``(N,2)`` ``float32`` numpy array. This function may change the input data in-place. rzBExpected from_shape and to_shape to not contain zeros. Got shapes z (from_shape) and z (to_shape).fN) r#allrr dtypekindr arrayastypefloat32) coords from_shapeto_shape from_height from_width to_heightto_widthvno_zeros_in_shapes coords_projs rproject_coords_r8>s*8!,Jx(H!A(1Q-' (1oK"1Q-Ix [*iJ KJqQUJ KL G0:H FG K >>& !V\\%6%6#%=hhv&--bjj9 $QT*Z78CK1$QT*[8IEK1  Ls C4cptj|rtj|}t |||S)aProject coordinates from one image shape to another. This performs a relative projection, e.g. a point at ``60%`` of the old image width will be at ``60%`` of the new image width after projection. Parameters ---------- coords : ndarray or list of tuple of number Coordinates to project. Either an ``(N,2)`` numpy array or a ``list`` containing ``(x,y)`` coordinate ``tuple`` s. from_shape : tuple of int or ndarray Old image shape. to_shape : tuple of int or ndarray New image shape. Returns ------- ndarray Projected coordinates as ``(N,2)`` ``float32`` numpy array. )rr r r r8)r.r/r0s rproject_coordsr:rs,2 ~~f 6:x 88rc|dkrgS|\}}|\}}tj||z ||z g}|d|zz }tj|D cgc]} || dz|dzz|| dz|dzzf c} Scc} w)aInterpolate ``N`` points on a line segment. Parameters ---------- point_a : iterable of number Start point of the line segment, given as ``(x,y)`` coordinates. point_b : iterable of number End point of the line segment, given as ``(x,y)`` coordinates. nb_steps : int Number of points to interpolate between `point_a` and `point_b`. Returns ------- list of tuple of number The interpolated points. Does not include `point_a`. r'r)r r-smxrange) point_apoint_bnb_stepsx1y1x2y2vec step_sizeis rinterpolate_point_pairrHs*!| FB FB **b2grBw' (Cq8|$I 99X     q1u ! $ $bAEYq\+A&AB    s#A4c t|dkr|S|rt||dgz}g}t|dd|ddD]%\}}|j|gt |||z'|s|j |d|S)a7Interpolate ``N`` on each line segment in a line string. Parameters ---------- points : iterable of iterable of number Points on the line segments, each one given as ``(x,y)`` coordinates. They are assumed to form one connected line string. nb_steps : int Number of points to interpolate on each individual line string. closed : bool, optional If ``True`` the output contains the last point in `points`. Otherwise it does not (but it will contain the interpolated points leading to the last point). Returns ------- list of tuple of number Coordinates of `points`, with additional `nb_steps` new points interpolated between each point pair. If `closed` is ``False``, the last point in `points` is not returned. r'rN)lenrzipextendrHr )pointsr@closed points_interpr>r?s rinterpolate_pointsrQs2 6{a  f +Ms VABZ8 I$Wgx@ A 9 VBZ( rc |dkDs Jd|fzt|dkr|S|rt||dgz}g}t|dd|ddD]f\}}tj|d|dz dz|d|dz dzz}t ||z dz }|j |gt|||zh|s|j|d|S)aInterpolate points with distance ``d`` on a line string. For a list of points ``A, B, C``, if the distance between ``A`` and ``B`` is greater than `max_distance`, it will place at least one point between ``A`` and ``B`` at ``A + max_distance * (B - A)``. Multiple points can be placed between the two points if they are far enough away from each other. The process is repeated for ``B`` and ``C``. Parameters ---------- points : iterable of iterable of number Points on the line segments, each one given as ``(x,y)`` coordinates. They are assumed to form one connected line string. max_distance : number Maximum distance between any two points in the result. closed : bool, optional If ``True`` the output contains the last point in `points`. Otherwise it does not (but it will contain the interpolated points leading to the last point). Returns ------- list of tuple of number Coordinates of `points`, with interpolated points added to the iterable. If `closed` is ``False``, the last point in `points` is not returned. rz3Expected max_distance to have a value >0, got %.8f.r'NrJr%) rKrrLr sqrtintrMrHr )rN max_distancerOrPr>r?distr@s r"interpolate_points_by_max_distancerWs> ! = A   6{a  f +Ms VABZ8ww QZ'!* $ *qzGAJ&1, -.|+q01 I$Wgx@ A B 9 VBZ( rct|ts|jSg}|D]!}|j|j#|S)aConvert coordinate-based augmentables to KeypointsOnImage instances. Added in 0.4.0. Parameters ---------- cbaois : list of imgaug.augmentables.bbs.BoundingBoxesOnImage or list of imgaug.augmentables.bbs.PolygonsOnImage or list of imgaug.augmentables.bbs.LineStringsOnImage or imgaug.augmentables.bbs.BoundingBoxesOnImage or imgaug.augmentables.bbs.PolygonsOnImage or imgaug.augmentables.bbs.LineStringsOnImage Coordinate-based augmentables to convert, e.g. bounding boxes. Returns ------- list of imgaug.augmentables.kps.KeypointsOnImage or imgaug.augmentables.kps.KeypointsOnImage ``KeypointsOnImage`` instances containing the coordinates of input `cbaois`. )rrto_keypoints_on_imager )cbaoiskpsoiscbaois rconvert_cbaois_to_kpsoisr]sF" fd #++-- F e1134 Mrct|ts?t|trJdt|jzf|j |Sg}t ||D]'\}}|j |}|j |)|S)aInvert the output of :func:`convert_to_cbaois_to_kpsois` in-place. This function writes in-place into `cbaois`. Added in 0.4.0. Parameters ---------- cbaois : list of imgaug.augmentables.bbs.BoundingBoxesOnImage or list of imgaug.augmentables.bbs.PolygonsOnImage or list of imgaug.augmentables.bbs.LineStringsOnImage or imgaug.augmentables.bbs.BoundingBoxesOnImage or imgaug.augmentables.bbs.PolygonsOnImage or imgaug.augmentables.bbs.LineStringsOnImage Original coordinate-based augmentables before they were converted, i.e. the same inputs as provided to :func:`convert_to_kpsois`. kpsois : list of imgaug.augmentables.kps.KeypointsOnImages or imgaug.augmentables.kps.KeypointsOnImages Keypoints to convert back to the types of `cbaois`, i.e. the outputs of :func:`convert_cbaois_to_kpsois`. Returns ------- list of imgaug.augmentables.bbs.BoundingBoxesOnImage or list of imgaug.augmentables.bbs.PolygonsOnImage or list of imgaug.augmentables.bbs.LineStringsOnImage or imgaug.augmentables.bbs.BoundingBoxesOnImage or imgaug.augmentables.bbs.PolygonsOnImage or imgaug.augmentables.bbs.LineStringsOnImage Parameter `cbaois`, with updated coordinates and shapes derived from `kpsois`. `cbaois` is modified in-place. zFExpected non-list for `kpsois` when `cbaois` is non-list. Got type %s.)rrr!__name__invert_to_keypoints_on_image_rLr )rZr[rr\kpsoicbaoi_recovereds r invert_convert_cbaois_to_kpsois_rc+s0 fd #fd+ 7 "6??3 5.7 7+33F;; FFF+ u==eD o&, Mrc|jDcgc]"}|j|j|kr|$c}|_|Scc}wr)itemscompute_out_of_image_fractionr")r\fractionitems r_remove_out_of_image_fraction_riRsJG$  - -ekk :X E GEK LGs'Ac t||||fDcgc]}|duc}rLtjd|d|d|d|d d||nd }||nd }||nd }||nd }||z|z }||z|z }||fScc}w) zCNormalize ``shift()`` arguments to x, y and handle deprecated args.Nz Got one of the arguments `top` (z ), `right` (z ), `bottom` (z ), `left` (zC) in a shift() call. These are deprecated. Use `x` and `y` instead.) stacklevelr)anyrwarn_deprecated)xytoprightbottomleftr5s r_normalize_shift_argsruZs C#= >#=aATM#= >? UFD *   _c!*!-1'tQ Hu  Gf  a4K ?s A0)T)NNNN)__doc__ __future__rrrr rnumpyr six.movesmovesr<imgaugrrrr#r8r:rHrQrWr]rcrirurrr}s\4@@  !"*1h9@ B'T1h4#Nr