o eN@sdZddlmZmZmZddlZddlZddl m Z ddl m ZddlmZddlmZmZmZdd d ZGd d d eZGdddeZdS)zDClasses to represent keypoints, i.e. points given as xy-coordinates.)print_functiondivisionabsolute_importN)imgaug) IAugmentable)normalize_shapeproject_coords_remove_out_of_image_fraction_h㈵>cCs"|dur|dus Jtd|}t|d} tjj||g}|dkdddf}d||}t|}||}t|||d} t |t|} | dkrT| } n.| t |kr\|S| ||} tj | } | dkrndn| | }t dd|| t d||} tjj|| |kr| S| }q)aEstimate the geometric median of points in 2D. Code from https://stackoverflow.com/a/30305181 Parameters ---------- points : (N,2) ndarray Points in 2D. Second axis must be given in xy-form. eps : float, optional Distance threshold when to return the median. X : None or (N,2) ndarray, optional Deprecated. Returns ------- (2,) ndarray Geometric median as xy-coordinate. Nz.Using 'X' is deprecated, use 'points' instead.rTr)iaZwarn_deprecatednpmeanscipyZspatialZdistanceZcdistsumlenZlinalgZnormmaxminZ euclidean)pointsZepsXydistZnonzerosZdist_invZ dist_inv_sumZ dist_inv_normTZ num_zerosy1RrZrinvrGD:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/kps.pycompute_geometric_medians2         rc@seZdZdZddZeddZeddZedd Zed d Z ed d Z ddZ ddZ ddZ ddZd2ddZd2ddZ  d3d d!Z d4d"d#Zd5d%d&Zd5d'd(Zd6d*d+Zd6d,d-Zd.d/Zd0d1Zd)S)7KeypointzA single keypoint (aka landmark) on an image. Parameters ---------- x : number Coordinate of the keypoint on the x axis. y : number Coordinate of the keypoint on the y axis. cCs||_||_dSNxrselfr#rrrr__init__Ss zKeypoint.__init__cCs,tjdtjd}|j|jg|dddf<|S)zGet the xy-coordinates as an ``(N,2)`` ndarray. Added in 0.4.0. Returns ------- ndarray An ``(N, 2)`` ``float32`` ndarray with ``N=1`` containing the coordinates of this keypoints. )rrdtyperN)remptyfloat32r#r)r%ZarrrrrcoordsWs zKeypoint.coordscCtt|jS)zGet the keypoint's x-coordinate, rounded to the closest integer. Returns ------- result : int Keypoint's x-coordinate, rounded to the closest integer. )intrroundr#r%rrrx_inth zKeypoint.x_intcCr,)zGet the keypoint's y-coordinate, rounded to the closest integer. Returns ------- result : int Keypoint's y-coordinate, rounded to the closest integer. )r-rr.rr/rrry_inttr1zKeypoint.y_intcCs|jdddfS)zGet the keypoint's x- and y-coordinate as a single array. Added in 0.4.0. Returns ------- ndarray A ``(2,)`` ``ndarray`` denoting the xy-coordinate pair. rN)r+r/rrrxys z Keypoint.xycCst|jtjS)zGet the keypoint's xy-coord, rounded to closest integer. Added in 0.4.0. Returns ------- ndarray A ``(2,)`` ``ndarray`` denoting the xy-coordinate pair. )rr.r3astypeZint32r/rrrxy_ints zKeypoint.xy_intcCs*t|j|jfg||}|d\|_|_|S)aProject in-place the keypoint onto a new position on a new image. E.g. if the keypoint is on its original image at ``x=(10 of 100 pixels)`` and ``y=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(20, 40)``. 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.kps.Keypoint Keypoint object with new coordinates. The instance of the keypoint may have been modified in-place. r)r r#r)r% from_shapeto_shapeZxy_projrrrproject_szKeypoint.project_cC|||S)a$Project the keypoint onto a new position on a new image. E.g. if the keypoint is on its original image at ``x=(10 of 100 pixels)`` and ``y=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(20, 40)``. 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.kps.Keypoint Keypoint object with new coordinates. )deepcopyr8)r%r6r7rrrprojectszKeypoint.projectcCsXt|}|dd\}}d|jko|kn}d|jko#|kn}| p+| S)aEstimate whether this point is outside of the given 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 ------- bool ``True`` is the point is inside the image plane, ``False`` otherwise. rr)r rr#)r%imageshapeheightwidthZy_insideZx_insiderrris_out_of_images  zKeypoint.is_out_of_imagecCst||S)aCompute fraction of the keypoint that is out of the image plane. The fraction is always either ``1.0`` (point is outside of the image plane) or ``0.0`` (point is inside the image plane). This method exists for consistency with other augmentables, e.g. bounding boxes. 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 Either ``1.0`` (point is outside of the image plane) or ``0.0`` (point is inside of it). )floatr@r%r<rrrcompute_out_of_image_fractionz&Keypoint.compute_out_of_image_fractionrcCs |j|7_|j|7_|S)aMove the keypoint around on an image in-place. Added in 0.4.0. Parameters ---------- x : number, optional Move by this value on the x axis. y : number, optional Move by this value on the y axis. Returns ------- imgaug.augmentables.kps.Keypoint Keypoint object with new coordinates. The instance of the keypoint may have been modified in-place. r"r$rrrshift_ szKeypoint.shift_cCr9)aiMove the keypoint around on an image. Parameters ---------- x : number, optional Move by this value on the x axis. y : number, optional Move by this value on the y axis. Returns ------- imgaug.augmentables.kps.Keypoint Keypoint object with new coordinates. r:rEr$rrrshift#szKeypoint.shiftrr?TFcCsj|rt|}|jdkrt|sJdt|fn|jdkr-t|r-|g|jd}|j}|}|dkr8|S|dkr?d}n|jtj dd }|t |}|jd d\} } |j |j } } t | |dd } t| d|d| }t | |dd }t| d|d| }t| |gd | \}}t||gd | \}}|d kp|| k}|d kp|| dk}|d kp|| k}|d kp|| dk}|o|}|o|}||dk}||dk}|s|s|s|s|dkr||||||f<n&d||||||f||||||f<n |rtd | | |jf|jj|jkr3|jd kr,tj|d d |d}|j|dd }|S)a*Draw the keypoint onto a given image. The keypoint is drawn as a square. Parameters ---------- image : (H,W,3) ndarray The image onto which to draw the keypoint. color : int or list of int or tuple of int or (3,) ndarray, optional The RGB color of the keypoint. If a single ``int`` ``C``, then that is equivalent to ``(C,C,C)``. alpha : float, optional The opacity of the drawn keypoint, where ``1.0`` denotes a fully visible keypoint and ``0.0`` an invisible one. size : int, optional The size of the keypoint. If set to ``S``, each square will have size ``S x S``. copy : bool, optional Whether to copy the image before drawing the keypoint. raise_if_out_of_image : bool, optional Whether to raise an exception if the keypoint is outside of the image. Returns ------- image : (H,W,3) ndarray Image with drawn keypoint. rzHGot a 2D image. Expected then 'color' to be a single number, but got %s.rKg{Gz?gGz?rF)copyrz;Cannot draw keypoint x=%.8f, y=%.8f on image with shape %s.uint8rI)out)rrMndimr Zis_single_numberstrr=r(r4r*arrayr2r0rrclip Exceptionname)r%r<coloralphasizerMraise_if_out_of_imageZ input_dtypeZ alpha_colorr>r?rr#x1x2ry2Z x1_clippedZ x2_clippedZ y1_clippedZ y2_clippedZx1_clipped_ooiZx2_clipped_ooiZy1_clipped_ooiZy2_clipped_ooiZx_ooiZy_ooiZ x_zero_sizeZ y_zero_sizerrr draw_on_image6sl%       zKeypoint.draw_on_imagec s|d|d|d}tj|dftjd}tj||j|||d|}d}d}t|D]B\} } |dkr@jg} ntj|dd|j|dd||} | D] } | | g||<|d7}q[| |krq|d7}q3|d8}q3|rz|Sfdd|DS)aGenerate nearby points based on manhattan distance. To generate the first neighbouring points, a distance of ``S`` (step size) is moved from the center point (this keypoint) to the top, right, bottom and left, resulting in four new points. From these new points, the pattern is repeated. Overlapping points are ignored. The resulting points have a shape similar to a square rotated by ``45`` degrees. Parameters ---------- nb_steps : int The number of steps to move from the center point. ``nb_steps=1`` results in a total of ``5`` output points (one center point + four neighbours). step_size : number The step size to move from every point to its neighbours. return_array : bool, optional Whether to return the generated points as a list of :class:`Keypoint` or an array of shape ``(N,2)``, where ``N`` is the number of generated points and the second axis contains the x-/y-coordinates. Returns ------- list of imgaug.augmentables.kps.Keypoint or (N,2) ndarray If `return_array` was ``False``, then a list of :class:`Keypoint`. Otherwise a numpy array of shape ``(N,2)``, where ``N`` is the number of generated points and the second axis contains the x-/y-coordinates. The center keypoint (the one on which this function was called) is always included. rrr'rcs"g|] }j|d|ddqSrrr"r:).0pointr/rr s"z>Keypoint.generate_similar_points_manhattan..)rzerosr*Zlinspacer enumerater#) r%Znb_stepsZ step_sizeZ return_arrayZ nb_pointsryyr?Z nth_pointZi_yrxxr#rr/r!generate_similar_points_manhattans4.        z*Keypoint.generate_similar_points_manhattan-C6?cCsjt|r |j}nt|rtt|}nt|ts$Jdt|f|j j}|j }t j |j||ddS)aEstimate if this and another KP have almost identical coordinates. Added in 0.4.0. Parameters ---------- other : imgaug.augmentables.kps.Keypoint or iterable The other keypoint with which to compare this one. If this is an ``iterable``, it is assumed to contain the xy-coordinates of a keypoint. max_distance : number, optional The maximum euclidean distance between a this keypoint and the other one. If the distance is exceeded, the two keypoints are not viewed as equal. Returns ------- bool Whether the two keypoints have almost identical coordinates. zcExpected 'other' to be an iterable containing one (x,y)-coordinate pair or a Keypoint. Got type %s.r)ZatolZrtol) r Z is_np_arrayZflat is_iterablelistflatten isinstancer typer+rZallclose)r%other max_distanceZcoords_bZcoords_arrrcoords_almost_equalss   zKeypoint.coords_almost_equalscC|j||dS)aCompare this and another KP's coordinates. .. note:: This method is currently identical to ``coords_almost_equals``. It exists for consistency with ``BoundingBox`` and ``Polygons``. Added in 0.4.0. Parameters ---------- other : imgaug.augmentables.kps.Keypoint or iterable The other object to compare against. Expected to be a ``Keypoint``. max_distance : number, optional See :func:`~imgaug.augmentables.kps.Keypoint.coords_almost_equals`. Returns ------- bool ``True`` if the coordinates are almost equal. Otherwise ``False``. )ro)rp)r%rnrorrr almost_equalsszKeypoint.almost_equalsNcCrq)aCreate a shallow copy of the keypoint instance. Parameters ---------- x : None or number, optional Coordinate of the keypoint on the x axis. If ``None``, the instance's value will be copied. y : None or number, optional Coordinate of the keypoint on the y axis. If ``None``, the instance's value will be copied. Returns ------- imgaug.augmentables.kps.Keypoint Shallow copy. r"r_r$rrrrM,sz Keypoint.copycCs0|dur|jn|}|dur|jn|}t||dS)aCreate a deep copy of the keypoint instance. Parameters ---------- x : None or number, optional Coordinate of the keypoint on the x axis. If ``None``, the instance's value will be copied. y : None or number, optional Coordinate of the keypoint on the y axis. If ``None``, the instance's value will be copied. Returns ------- imgaug.augmentables.kps.Keypoint Deep copy. Nr")r#rr r$rrrr:As zKeypoint.deepcopycC|Sr!__str__r/rrr__repr__XzKeypoint.__repr__cCsd|j|jfS)NzKeypoint(x=%.8f, y=%.8f)r"r/rrrru[szKeypoint.__str__rrrHrJrKTFF)rhNN)__name__ __module__ __qualname____doc__r&propertyr+r0r2r3r5r8r;r@rCrErGr]rgrprrrMr:rvrurrrrr Fs<           e O '   r c@seZdZdZddZeddZejddZeddZed d Z ed d Z d dZ ddZ  dPddZ ddZddZddZddZdQd!d"ZdQd#d$Zejd%d&d'd(Zd)d*Zeejd+d&d,d-Zed.d/Zd0d1ZdRd3d4Zed5d5d6d2d7fd8d9ZdSd:d;Zedd5d5d6d7d7fdd?Z!d@dAZ"dTdBdCZ#dTdDdEZ$dFdGZ%dHdIZ&dJdKZ'dLdMZ(dNdOZ)d7S)UKeypointsOnImageaContainer for all keypoints on a single image. Parameters ---------- keypoints : list of imgaug.augmentables.kps.Keypoint List of keypoints 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.kps import Keypoint, KeypointsOnImage >>> >>> image = np.zeros((70, 70)) >>> kps = [Keypoint(x=10, y=20), Keypoint(x=34, y=60)] >>> kps_oi = KeypointsOnImage(kps, shape=image.shape) cCs||_t||_dSr!) keypointsr r=r%rr=rrrr&wszKeypointsOnImage.__init__cCs|jS)zGet the keypoints in this container. Added in 0.4.0. Returns ------- list of Keypoint Keypoints within this container. rr/rrritems{s zKeypointsOnImage.itemscCs ||_dS)zSet the keypoints in this container. Added in 0.4.0. Parameters ---------- value : list of Keypoint Keypoints within this container. Nr)r%valuerrrrs cC |jdS)zfGet the image height. Returns ------- int Image height. rr=r/rrrr> zKeypointsOnImage.heightcCr)zdGet the image width. Returns ------- int Image width. rrr/rrrr?rzKeypointsOnImage.widthcCst|jdkS)zDetermine whether this object contains zero keypoints. Returns ------- bool ``True`` if this object contains zero keypoints. r)rrr/rrrr)s zKeypointsOnImage.emptycCs^t|}|dd|jddkr||_|St|jD]\}}||j||j|<q||_|S)aProject all keypoints 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 keypoints are to be projected. May also simply be that new image's shape tuple. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Object containing all projected keypoints. The object may have been modified in-place. rr)r r=rdrr8)r%r<Zon_shapeikprrron_szKeypointsOnImage.on_cC||S)aProject all keypoints from one image shape to a new one. Parameters ---------- image : ndarray or tuple of int New image onto which the keypoints are to be projected. May also simply be that new image's shape tuple. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Object containing all projected keypoints. )r:rrBrrronszKeypointsOnImage.onrHrJrKTFc Cs8|rt|n|}|jD] }|j||||d|d}q |S)aDraw all keypoints onto a given image. Each keypoint is drawn as a square of provided color and size. Parameters ---------- image : (H,W,3) ndarray The image onto which to draw the keypoints. This image should usually have the same shape as set in ``KeypointsOnImage.shape``. color : int or list of int or tuple of int or (3,) ndarray, optional The RGB color of all keypoints. If a single ``int`` ``C``, then that is equivalent to ``(C,C,C)``. alpha : float, optional The opacity of the drawn keypoint, where ``1.0`` denotes a fully visible keypoint and ``0.0`` an invisible one. size : int, optional The size of each point. If set to ``C``, each square will have size ``C x C``. copy : bool, optional Whether to copy the image before drawing the points. raise_if_out_of_image : bool, optional Whether to raise an exception if any keypoint is outside of the image. Returns ------- (H,W,3) ndarray Image with drawn keypoints. F)rVrWrXrMrY)rrMrr])r%r<rVrWrXrMrYkeypointrrrr]s'  zKeypointsOnImage.draw_on_imagecCs t||S)aRemove all KPs with an OOI fraction of at least `fraction` in-place. 'OOI' is the abbreviation for 'out of image'. This method exists for consistency with other augmentables, e.g. bounding boxes. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a keypoint has to have in order to be removed. Note that any keypoint can only have a fraction of either ``1.0`` (is outside) or ``0.0`` (is inside). Set this to ``0.0+eps`` to remove all points that are outside of the image. Setting this to ``0.0`` will remove all points. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Reduced set of keypoints, with those thathad an out of image fraction greater or equal the given one removed. The object may have been modified in-place. )r r%fractionrrrremove_out_of_image_fraction_s z.KeypointsOnImage.remove_out_of_image_fraction_cCr)aSRemove all KPs with an out of image fraction of at least `fraction`. This method exists for consistency with other augmentables, e.g. bounding boxes. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a keypoint has to have in order to be removed. Note that any keypoint can only have a fraction of either ``1.0`` (is outside) or ``0.0`` (is inside). Set this to ``0.0+eps`` to remove all points that are outside of the image. Setting this to ``0.0`` will remove all points. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Reduced set of keypoints, with those thathad an out of image fraction greater or equal the given one removed. )r:rrrrrremove_out_of_image_fraction5rDz-KeypointsOnImage.remove_out_of_image_fractioncCs |dS)ayRemove all KPs that are outside of the image plane. This method exists for consistency with other augmentables, e.g. bounding boxes. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Keypoints that are inside the image plane. The object may have been modified in-place. ?)rr/rrrclip_out_of_image_Os z#KeypointsOnImage.clip_out_of_image_cCs |S)aARemove all KPs that are outside of the image plane. This method exists for consistency with other augmentables, e.g. bounding boxes. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Keypoints that are inside the image plane. )r:rr/rrrclip_out_of_imageas z"KeypointsOnImage.clip_out_of_imagercCs,t|jD]\}}|j||d|j|<q|S)aMove the keypoints on the x/y-axis in-place. Added in 0.4.0. Parameters ---------- x : number, optional Move each keypoint by this value on the x axis. y : number, optional Move each keypoint by this value on the y axis. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Keypoints after moving them. The object and its items may have been modified in-place. r")rdrrE)r%r#rrrrrrrEqszKeypointsOnImage.shift_cCs|j||dS)aMove the keypoints on the x/y-axis. Parameters ---------- x : number, optional Move each keypoint by this value on the x axis. y : number, optional Move each keypoint by this value on the y axis. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Keypoints after moving them. r"rFr$rrrrGszKeypointsOnImage.shiftzKeypointsOnImage.to_xy_array())Zalt_funccCrs)-Convert all keypoint coordinates to an array of shape ``(N,2)``. Returns ------- (N, 2) ndarray Array containing the coordinates of all keypoints. ``N`` denotes the number of keypoints. The second axis denotes the x/y-coordinates. ) to_xy_arrayr/rrrget_coords_arrays z!KeypointsOnImage.get_coords_arraycCsNtjt|jdftjd}t|jD]\}}|j||df<|j||df<q|S)rrr'rr)rrcrrr*rdr#r)r%resultrrrrrrs  zKeypointsOnImage.to_xy_arrayz KeypointsOnImage.from_xy_array()cCs t||S)aConvert an ``(N,2)`` array to a ``KeypointsOnImage`` object. Parameters ---------- coords : (N, 2) ndarray Coordinates of ``N`` keypoints on an image, given as a ``(N,2)`` array of xy-coordinates. shape : tuple The shape of the image on which the keypoints are placed. Returns ------- imgaug.augmentables.kps.KeypointsOnImage :class:`KeypointsOnImage` object containing the array's keypoints. )r from_xy_array)r+r=rrrfrom_coords_arrays z"KeypointsOnImage.from_coords_arraycCshtj|tjd}|jddkrtg|S|jdkr |jddks(Jd|jfdd|D}t||S)a-Convert an ``(N,2)`` array to a ``KeypointsOnImage`` object. Parameters ---------- xy : (N, 2) ndarray or iterable of iterable of number Coordinates of ``N`` keypoints on an image, given as a ``(N,2)`` array of xy-coordinates. shape : tuple of int or ndarray The shape of the image on which the keypoints are placed. Returns ------- imgaug.augmentables.kps.KeypointsOnImage :class:`KeypointsOnImage` object containing the array's keypoints. r'rrrL7Expected input array to have shape (N,2), got shape %s.cSs g|] }t|d|ddqSr^)r )r`coordrrrrbs z2KeypointsOnImage.from_xy_array..)rrRr*r=rrP)clsr3r=rrrrrs  zKeypointsOnImage.from_xy_arraycCstj|tjd}|jddks#|jdkr|jddks#Jd|jft|t|jks9Jdt|t|jft|j|D] \}\}}||_||_ q?|S)a Modify the keypoint coordinates of this instance in-place. .. note:: This currently expects that `xy` contains exactly as many coordinates as there are keypoints in this instance. Otherwise, an ``AssertionError`` will be raised. Added in 0.4.0. Parameters ---------- xy : (N, 2) ndarray or iterable of iterable of number Coordinates of ``N`` keypoints on an image, given as a ``(N,2)`` array of xy-coordinates. ``N`` must match the number of keypoints in this instance. Returns ------- KeypointsOnImage This instance itself, with updated keypoint coordinates. Note that the instance was modified in-place. r'rrrLrzxExpected to receive as many keypoint coordinates as there are currently keypoints in this instance. Got %d, expected %d.) rrRr*r=rPrrzipr#r)r%r3rr#rrrrfill_from_xy_array_s (z$KeypointsOnImage.fill_from_xy_array_rcCs:|jdd\}}tj||t|jftjd}|ddks$Jd|ftd|dd}t|jD]h\}}|j}|j } t | |d|d} t | |dd|} t ||d|d} t ||dd|} | | kr{| | kr{d|| | | | |f<d|kr|krnq2d| kr|krnq2d||| |f<q2|S)a|Create an ``(H,W,N)`` image with keypoint coordinates set to ``255``. This method generates a new ``uint8`` array of shape ``(H,W,N)``, where ``H`` is the ``.shape`` height, ``W`` the ``.shape`` width and ``N`` is the number of keypoints. The array is filled with zeros. The coordinate of the ``n``-th keypoint is set to ``255`` in the ``n``-th channel. This function can be used as a helper when augmenting keypoints with a method that only supports the augmentation of images. Parameters ------- size : int Size of each (squared) point. Returns ------- (H,W,N) ndarray Image in which the keypoints are marked. ``H`` is the height, defined in ``KeypointsOnImage.shape[0]`` (analogous ``W``). ``N`` is the number of keypoints. rrr'z5Expected 'size' to have an odd value, got %d instead.rrI) r=rrcrrrNrrdr2r0rS)r%rXr>r?r<Zsizehrrrr#rZr[rr\rrrto_keypoint_images&0z"KeypointsOnImage.to_keypoint_imagerLr"NcCsn|jdksJd|j|jf|j\}}}d}|dur"d}d}d} n6t|ttfrAt|dks8Jdt|f|d }|d } nt|trO|d }|d } n td t|fg} t |D]D} t |d| f} t | ||f} || d | d | f|k}|r| d d}| d d}| t||dq_|rq_| t|| dq_||f}|dur||f7}t| |dS)aConvert ``to_keypoint_image()`` outputs to ``KeypointsOnImage``. This is the inverse of :func:`KeypointsOnImage.to_keypoint_image`. Parameters ---------- image : (H,W,N) ndarray The keypoints image. N is the number of keypoints. if_not_found_coords : tuple or list or dict or None, optional Coordinates to use for keypoints that cannot be found in `image`. * If this is a ``list``/``tuple``, it must contain two ``int`` values. * If it is a ``dict``, it must contain the keys ``x`` and ``y`` with each containing one ``int`` value. * If this is ``None``, then the keypoint will not be added to the final :class:`KeypointsOnImage` object. threshold : int, optional The search for keypoints works by searching for the argmax in each channel. This parameters contains the minimum value that the max must have in order to be viewed as a keypoint. nb_channels : None or int, optional Number of channels of the image on which the keypoints are placed. Some keypoint augmenters require that information. If set to ``None``, the keypoint's shape will be set to ``(height, width)``, otherwise ``(height, width, nb_channels)``. Returns ------- imgaug.augmentables.kps.KeypointsOnImage The extracted keypoints. rKzHExpected 'image' to have three dimensions, got %d with shape %s instead.FNTrLrzRExpected tuple 'if_not_found_coords' to contain exactly two values, got %d values.rrr#rIExpected if_not_found_coords to be None or tuple or list or dict, got %s..rr"r)rPr=rltuplerjrdictrTrmsmxrangerargmax unravel_indexappendr r)r<if_not_found_coords threshold nb_channelsr>r? nb_keypointsdrop_if_not_foundif_not_found_xif_not_found_yrrZ maxidx_flatZ maxidx_ndimfoundr#r out_shaperrrfrom_keypoint_imageKsZ (          z$KeypointsOnImage.from_keypoint_imagec Cs|jdd\}}tj||t|jftjd}td|}td|}t||\}}t|jD] \} } | j | j } } || d|| d|dddd| f<q0t |}|r^d|dS|S)aGenerate a ``(H,W,N)`` array of distance maps for ``N`` keypoints. The ``n``-th distance map contains at every location ``(y, x)`` the euclidean distance to the ``n``-th keypoint. This function can be used as a helper when augmenting keypoints with a method that only supports the augmentation of images. Parameters ------- inverted : bool, optional If ``True``, inverted distance maps are returned where each distance value d is replaced by ``d/(d+1)``, i.e. the distance maps have values in the range ``(0.0, 1.0]`` with ``1.0`` denoting exactly the position of the respective keypoint. Returns ------- (H,W,N) ndarray A ``float32`` array containing ``N`` distance maps for ``N`` keypoints. Each location ``(y, x, n)`` in the array denotes the euclidean distance at ``(y, x)`` to the ``n``-th keypoint. If `inverted` is ``True``, the distance ``d`` is replaced by ``d/(d+1)``. The height and width of the array match the height and width in ``KeypointsOnImage.shape``. rrr'Nr) r=rrcrrr*ZarangeZmeshgridrdrr#sqrt) r%invertedr>r? distance_mapsrerfZgrid_xxZgrid_yyrrrr#rrrto_distance_mapss  ,  z!KeypointsOnImage.to_distance_mapscCs|jdksJd|j|jf|j\}}}d}|dur"d}d} d} n6t|ttfrAt|dks8Jdt|f|d } |d } nt|trO|d } |d } n td t|fg} t |D]e} |rmt |d| f} n t |d| f} t | ||f}|s|dur||d |d | f|k}n|r|dur||d |d | f|k}nd}|r| t|d |d dq_|rq_| t| | dq_||f}|dur||f7}t| |dS)a_Convert outputs of ``to_distance_maps()`` to ``KeypointsOnImage``. This is the inverse of :func:`KeypointsOnImage.to_distance_maps`. Parameters ---------- distance_maps : (H,W,N) ndarray The distance maps. ``N`` is the number of keypoints. inverted : bool, optional Whether the given distance maps were generated in inverted mode (i.e. :func:`KeypointsOnImage.to_distance_maps` was called with ``inverted=True``) or in non-inverted mode. if_not_found_coords : tuple or list or dict or None, optional Coordinates to use for keypoints that cannot be found in `distance_maps`. * If this is a ``list``/``tuple``, it must contain two ``int`` values. * If it is a ``dict``, it must contain the keys ``x`` and ``y`` with each containing one ``int`` value. * If this is ``None``, then the keypoint will not be added to the final :class:`KeypointsOnImage` object. threshold : float, optional The search for keypoints works by searching for the argmin (non-inverted) or argmax (inverted) in each channel. This parameters contains the maximum (non-inverted) or minimum (inverted) value to accept in order to view a hit as a keypoint. Use ``None`` to use no min/max. nb_channels : None or int, optional Number of channels of the image on which the keypoints are placed. Some keypoint augmenters require that information. If set to ``None``, the keypoint's shape will be set to ``(height, width)``, otherwise ``(height, width, nb_channels)``. Returns ------- imgaug.augmentables.kps.KeypointsOnImage The extracted keypoints. rKzAExpected three-dimensional input, got %d dimensions and shape %s.FNTrLrzQExpected tuple/list 'if_not_found_coords' to contain exactly two entries, got %d.rrr#rr.r"r)rPr=rlrrjrrrTrmrrrrZargminrrr r)rrrrrr>r?rrrrrrZ hitidx_flatZ hitidx_ndimrrrrrfrom_distance_mapssj 1          z#KeypointsOnImage.from_distance_mapscCrs)a`Convert the keypoints to one ``KeypointsOnImage`` instance. This method exists for consistency with ``BoundingBoxesOnImage``, ``PolygonsOnImage`` and ``LineStringsOnImage``. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Copy of this keypoints instance. r_r/rrrto_keypoints_on_image9sz&KeypointsOnImage.to_keypoints_on_imagecCsbt|j}t|j|ksJd|t|jft|j|jD] \}}|j|_|j|_q|j|_|S)a-Invert the output of ``to_keypoints_on_image()`` in-place. This function writes in-place into this ``KeypointsOnImage`` instance. Added in 0.4.0. Parameters ---------- kpsoi : imgaug.augmentables.kps.KeypointsOnImages Keypoints to copy data from, i.e. the outputs of ``to_keypoints_on_image()``. Returns ------- KeypointsOnImage Keypoints container with updated coordinates. Note that the instance is also updated in-place. z Expected %d coordinates, got %d.)rrrr#rr=)r%ZkpsoiZ nb_points_expZ kp_targetZ kp_sourcerrrinvert_to_keypoints_on_image_Is   z.KeypointsOnImage.invert_to_keypoints_on_image_cCs2|dur |jdd}|durt|j}t||S)a.Create a shallow copy of the ``KeypointsOnImage`` object. Parameters ---------- keypoints : None or list of imgaug.Keypoint, optional List of keypoints on the image. If ``None``, the instance's keypoints will be copied. shape : tuple of int, optional The shape of the image on which the keypoints are placed. If ``None``, the instance's shape will be copied. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Shallow copy. Nrrr=rrrrrrMjs   zKeypointsOnImage.copycCs4|dur dd|jD}|durt|j}t||S)a(Create a deep copy of the ``KeypointsOnImage`` object. Parameters ---------- keypoints : None or list of imgaug.Keypoint, optional List of keypoints on the image. If ``None``, the instance's keypoints will be copied. shape : tuple of int, optional The shape of the image on which the keypoints are placed. If ``None``, the instance's shape will be copied. Returns ------- imgaug.augmentables.kps.KeypointsOnImage Deep copy. NcSsg|]}|qSrr_)r`rrrrrbsz-KeypointsOnImage.deepcopy..rrrrrr:s   zKeypointsOnImage.deepcopycCs |j|S)zGet the keypoint(s) with given indices. Added in 0.4.0. Returns ------- list of imgaug.augmentables.kps.Keypoint Keypoint(s) with given indices. r)r%indicesrrr __getitem__ zKeypointsOnImage.__getitem__cC t|jS)aIterate over the keypoints in this container. Added in 0.4.0. Yields ------ Keypoint A keypoint in this container. The order is identical to the order in the keypoint list provided upon class initialization. )iterrr/rrr__iter__s zKeypointsOnImage.__iter__cCr)zGet the number of items in this instance. Added in 0.4.0. Returns ------- int Number of items in this instance. )rrr/rrr__len__rzKeypointsOnImage.__len__cCrsr!rtr/rrrrvrwzKeypointsOnImage.__repr__cCsdt|j|jfS)NzKeypointsOnImage(%s, shape=%s))rQrr=r/rrrrus zKeypointsOnImage.__str__ryrx)rrzr{)*r|r}r~rr&rrsetterr>r?r)rrr]rrrrrErGr deprecatedrr staticmethodr classmethodrrrrrrrrrMr:rrrrvrurrrrr_sj       .         ,/ V- i !   r)Nr N)r __future__rrrnumpyrZscipy.spatial.distancerZ six.movesmovesrrr baserutilsr r r robjectr rrrrrs    8