o
    oi                     @   s  d Z ddlmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZ ddlmZ dededefddZdededefd	d
ZdededefddZdededefddZdededefddZdededefddZdededefddZdededefddZdededefddZdededefddZdededefddZdededefddZ dededefd d!Z!dededefd"d#Z"d$S )%zAutoAugment operation wrapper.    )AutoContrast
BrightnessContrastEqualizeInvertOperationBase	PosterizeRotateSaturate	SharpnessShearXShearYSolarize
TranslateX
TranslateY)linspaceprobability	magnitudereturnc                 C   8   t dddd }td| ||  ||d   fddS )	zReturn ShearX op.333333ӿ333333?        f@N   Fmagnitude_rangesymmetric_megnitude)r   r   itemr   r   
magnitudes r!   \/home/ubuntu/.local/lib/python3.10/site-packages/kornia/augmentation/auto/autoaugment/ops.pyshear_x(      r#   c                 C   r   )	zReturn ShearY op.r   r   r   r   Nr   Fr   )r   r   r   r   r!   r!   r"   shear_y3   r$   r%   c                 C   4   t ddd}td| ||  ||d   fddS )zReturn TranslateX op.            ?r   Nr   Fr   )r   r   r   r   r!   r!   r"   translate_x>      r)   c                 C   r&   )zReturn TranslateY op.r'   r(   r   Nr   Fr   )r   r   r   r   r!   r!   r"   translate_yI   r*   r+   c                 C   r&   )zReturn rotate op.i   r   Nr   Fr   )r   r	   r   r   r!   r!   r"   rotateT   r*   r-   _c                 C      t | S )zReturn AutoConstrast op.)r   r   r.   r!   r!   r"   auto_contrast_      r1   c                 C   r/   )zReturn invert op.)r   r0   r!   r!   r"   invertd   r2   r3   c                 C   r/   )zReturn equalize op.)r   r0   r!   r!   r"   equalizei   r2   r4   c                 C   s6   t dddd }td| ||  ||d   fdS )zReturn solarize op.r      r   g     o@Nr   r   )r   r   r   r   r!   r!   r"   solarizen   s   &r7   c                 C   2   t ddd}td| ||  ||d   fdS )zReturn posterize op.      r   Nr   r6   )r   r   r   r   r!   r!   r"   	posterizet      r;   c                 C   r8   )zReturn contrast op.皙?ffffff?r   Nr   r6   )r   r   r   r   r!   r!   r"   contrast|      &r?   c                 C   r8   )zReturn brightness op.r=   r>   r   Nr   r6   )r   r   r   r   r!   r!   r"   
brightness   r<   rA   c                 C   r8   )zReturn sharpness op.r=   r>   r   Nr   r6   )r   r   r   r   r!   r!   r"   	sharpness   r<   rB   c                 C   r8   )zReturn color op.r=   r>   r   Nr   r6   )r   r
   r   r   r!   r!   r"   color   r@   rC   N)#__doc__#kornia.augmentation.auto.operationsr   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   kornia.corer   floatintr#   r%   r)   r+   r-   r1   r3   r4   r7   r;   r?   rA   rB   rC   r!   r!   r!   r"   <module>   s"   D