o &j@sdZddlmZmZmZddlZddlZddlm Z ddl m Z ddl mZddl mZ hd Zd d Zd d ZddZddZddZddZGdddejZGdddejZdS)z Augmenters that apply mirroring/flipping operations to images. List of augmenters: * :class:`Fliplr` * :class:`Flipud` )print_functiondivisionabsolute_importN)_normalize_cv2_input_arr_)meta) parameters>Zint8Zint16Zuint8Zuint16cCs|jjtvr t|St|S)aFlip an image-like array horizontally. **Supported dtypes**: * ``uint8``: yes; fully tested * ``uint16``: yes; fully tested * ``uint32``: yes; fully tested * ``uint64``: yes; fully tested * ``int8``: yes; fully tested * ``int16``: yes; fully tested * ``int32``: yes; fully tested * ``int64``: yes; fully tested * ``float16``: yes; fully tested * ``float32``: yes; fully tested * ``float64``: yes; fully tested * ``float128``: yes; fully tested * ``bool``: yes; fully tested Parameters ---------- arr : ndarray A 2D/3D `(H, W, [C])` image array. Returns ------- ndarray Horizontally flipped array. Examples -------- >>> import numpy as np >>> import imgaug.augmenters.flip as flip >>> arr = np.arange(16).reshape((4, 4)) >>> arr_flipped = flip.fliplr(arr) Create a ``4x4`` array and flip it horizontally. )Zdtypename_FLIPLR_DTYPES_CV2 _fliplr_cv2_fliplr_slicedarrrW/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/imgaug/augmenters/flip.pyfliplrs +rcCs|ddddddfS)N.rrrrrr sr csjdkr tSjdkr-jddkr-fddtjdD}tj|dd}n j}|dssz_fliplr_cv2..)ZaxisZOWNDATAZ C_CONTIGUOUSrr.)sizenpcopyndimshapesmxrangestackflagsZascontiguousarrayrrrZnewaxis)rZchannelsresultr"rrrr s"     r cCs|ddddfS)aFlip an image-like array vertically. **Supported dtypes**: * ``uint8``: yes; fully tested * ``uint16``: yes; fully tested * ``uint32``: yes; fully tested * ``uint64``: yes; fully tested * ``int8``: yes; fully tested * ``int16``: yes; fully tested * ``int32``: yes; fully tested * ``int64``: yes; fully tested * ``float16``: yes; fully tested * ``float32``: yes; fully tested * ``float64``: yes; fully tested * ``float128``: yes; fully tested * ``bool``: yes; fully tested Parameters ---------- arr : ndarray A 2D/3D `(H, W, [C])` image array. Returns ------- ndarray Vertically flipped array. Examples -------- >>> import numpy as np >>> import imgaug.augmenters.flip as flip >>> arr = np.arange(16).reshape((4, 4)) >>> arr_flipped = flip.flipud(arr) Create a ``4x4`` array and flip it vertically. Nr.rrrrrflipuds)r$cOt|i|S)zAlias for Fliplr.)FliplrargskwargsrrrHorizontalFlip(r*cOr%)zAlias for Flipud.)Flipudr'rrr VerticalFlip.r+r-c8eZdZdZ   d fdd ZddZd d ZZS) r&a Flip/mirror input images horizontally. .. note:: The default value for the probability is ``0.0``. So, to flip *all* input images use ``Fliplr(1.0)`` and *not* just ``Fliplr()``. **Supported dtypes**: See :func:`~imgaug.augmenters.flip.fliplr`. Parameters ---------- p : number or imgaug.parameters.StochasticParameter, optional Probability of each image to get flipped. seed : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. name : None or str, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. random_state : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional Old name for parameter `seed`. Its usage will not yet cause a deprecation warning, but it is still recommended to use `seed` now. Outdated since 0.4.0. deterministic : bool, optional Deprecated since 0.4.0. See method ``to_deterministic()`` for an alternative and for details about what the "deterministic mode" actually does. Examples -------- >>> import imgaug.augmenters as iaa >>> aug = iaa.Fliplr(0.5) Flip ``50`` percent of all images horizontally. >>> aug = iaa.Fliplr(1.0) Flip all images horizontally. rN deprecatedc*tt|j||||dt|d|_dSN)seedr random_state deterministicp)superr&__init__iaphandle_probability_paramr5selfr5r2r r3r4 __class__rrr7e zFliplr.__init__cCs|jj|jf|d}t|D]\}}|dkr|jdur%t|j||j|<|jdur6t|j|j|j|_|jdurGt|j|j |j|_ |j durd|j |}|j d} |j D] } | t | j | _ qY|jdur|j|} | j d} | jD]} | j| j} }| || _| | | _qv|jdur|j|}|j d} |jD]}| |jdddf|jdddf<q|jdur|j|}|j d} |jD]}| |jdddf|jdddf<qq|S)Nr3?rr)r5 draw_samplesnb_rows enumerateimagesrheatmapsarr_0to1segmentation_mapsr keypointsrfloatxbounding_boxesx1x2polygonsexterior line_stringscoords)r;batchr3parentshookssamplesisamplekpsoiwidthkpbbsoibbrLrMpsoipolylsoilsrrr_augment_batch_nsN                    &    &zFliplr._augment_batch_cC|jgSz=See :func:`~imgaug.augmenters.meta.Augmenter.get_parameters`.r5r;rrrget_parameterszFliplr.get_parametersrNNr/r/__name__ __module__ __qualname____doc__r7rarf __classcell__rrr<rr&4s0 1r&cr.) r,aFlip/mirror input images vertically. .. note:: The default value for the probability is ``0.0``. So, to flip *all* input images use ``Flipud(1.0)`` and *not* just ``Flipud()``. **Supported dtypes**: See :func:`~imgaug.augmenters.flip.flipud`. Parameters ---------- p : number or imgaug.parameters.StochasticParameter, optional Probability of each image to get flipped. seed : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. name : None or str, optional See :func:`~imgaug.augmenters.meta.Augmenter.__init__`. random_state : None or int or imgaug.random.RNG or numpy.random.Generator or numpy.random.BitGenerator or numpy.random.SeedSequence or numpy.random.RandomState, optional Old name for parameter `seed`. Its usage will not yet cause a deprecation warning, but it is still recommended to use `seed` now. Outdated since 0.4.0. deterministic : bool, optional Deprecated since 0.4.0. See method ``to_deterministic()`` for an alternative and for details about what the "deterministic mode" actually does. Examples -------- >>> import imgaug.augmenters as iaa >>> aug = iaa.Flipud(0.5) Flip ``50`` percent of all images vertically. >>> aug = iaa.Flipud(1.0) Flip all images vertically. rNr/cr0r1)r6r,r7r8r9r5r:r<rrr7r>zFlipud.__init__cCs|jj|jf|d}t|D]\}}|dkr|jdur*|j|ddddf|j|<|jdur@|j|jddddf|j|_|jdurV|j|jddddf|j|_|j durs|j |}|j d} |j D] } | t | j | _ qh|j dur|j |} | j d} | j D]} | j| j} }| || _| | | _q|jdur|j|}|j d} |jD]}| |jdddf|jdddf<q|jdur|j|}|j d} |jD]}| |jdddf|jdddf<qq|S)Nr?r@r.rr)r5rArBrCrDrErFrGrrHrrIyrKy1y2rNrOrPrQ)r;rRr3rSrTrUrVrWrXheightrZr[r\rprqr]r^r_r`rrrrasJ                  &    &zFlipud._augment_batch_cCrbrcrdrerrrrfrgzFlipud.get_parametersrhrirrr<rr,s/ 3r,)rm __future__rrrnumpyrrZ six.movesZmovesrZ imgaug.imgaugrrr r8r rr r r$r*r-Z Augmenterr&r,rrrrs.     II0!,q