o e:/@sdZddlmZmZmZddlZddlZddl m Z ddl Z ddZddZdd Zd d Zd d ZddZdddZdddZddZddZddZdddZdS)z.Utility functions used in augmentable modules.)print_functionabsolute_importdivisionNcCsPt|r t|Sg}|D]}t|r|t|q||q|SN)ia is_np_arraynpcopyappenddeepcopy)Z augmentablesresultZ augmentabler ID:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/utils.pycopy_augmentables s   rcCs|durdSt|st|r|St|trdd|DSt|tr,tdd|DSt|r6t|St |dr?| St |S)NcSg|]}t|qSr  deepcopy_fast.0elr r r z!deepcopy_fast..cSrr rrr r rr rr ) rZis_single_numberZ is_string isinstancelisttuplerrr hasattrr copylib)objr r rrs      rcCs0t|tr|St|sJdt|f|jS)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 %s.)rrrrtypeshape)rr r rnormalize_shape)s   r c Cst|}t|}|dd|ddkr|S|dd\}}|dd\}}tdd||||fD}|s=Jd||f|}t|rJ|jjdkrSt|tj }|dddf|||dddf<|dddf|||dddf<|S) 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. rcSsg|]}|dkqS)rr rvr r rrcrz#project_coords_..zdExpected from_shape and to_shape to not contain zeros. Got shapes %s (from_shape) and %s (to_shape).fN) r allrrZdtypekindrarrayZastypefloat32) coords from_shapeto_shapeZ from_heightZ from_widthZ to_heightZto_widthZno_zeros_in_shapesZ coords_projr r rproject_coords_>s&$$r-cCs t|r t|}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. )rrrr r-)r*r+r,r r rproject_coordsrs   r.csZ|dkrgS|\|\}}t||g}|d|fddt|DS)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%cs4g|]}|dd|ddfqS)r%rr )riZ step_sizex1y1r rrs&z*interpolate_point_pair..)rr)smxrange)point_apoint_bnb_stepsZx2y2Zvecr r0rinterpolate_point_pairs r9TcCs|t|dkr|S|rt||dg}g}t|dd|ddD]\}}||gt|||q"|s<||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)lenrzipextendr9r )pointsr7closed points_interpr5r6r r rinterpolate_pointss " rAcCs|dks Jd|ft|dkr|S|rt||dg}g}t|dd|ddD]/\}}t|d|dd|d|dd}t||d}||gt|||q-|sf||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%Nr:r!) r;rr<rsqrtintr=r9r )r>Z max_distancer?r@r5r6distr7r r r"interpolate_points_by_max_distances2  " rEcCs2t|ts |Sg}|D] }||q |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`. )rrZto_keypoints_on_imager )cbaoiskpsoiscbaoir r rconvert_cbaois_to_kpsoiss rIcCsbt|tst|trJdt|jf||Sg}t||D]\}}||}||q |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__Zinvert_to_keypoints_on_image_r<r )rFrGr rHZkpsoiZcbaoi_recoveredr r r invert_convert_cbaois_to_kpsois_+s     rKcsfddjD_S)Ncs g|] }|jkr|qSr )Zcompute_out_of_image_fractionr)ritemrHfractionr rrSs z2_remove_out_of_image_fraction_..)itemsrMr rMr_remove_out_of_image_fraction_Rs rPcCstdd||||fDrFtjd||||fdd|dur |nd}|dur(|nd}|dur0|nd}|dur8|nd}|||}|||}||fS)zCNormalize ``shift()`` arguments to x, y and handle deprecated args.cSsg|]}|duqSrr r"r r rr\rz)_normalize_shift_args..zGot one of the arguments `top` (%s), `right` (%s), `bottom` (%s), `left` (%s) in a shift() call. These are deprecated. Use `x` and `y` instead.) stacklevelNr)anyrZwarn_deprecated)xytoprightbottomleftr r r_normalize_shift_argsZs  rZ)T)NNNN)__doc__ __future__rrrr rnumpyrZ six.movesmovesr3Zimgaugrrrr r-r.r9rArErIrKrPrZr r r rs$ 4 ! *4'