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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ededefd"d#Z!d$S )%zRandAugment operation wrapper.    )AutoContrast
BrightnessContrastEqualizeInvertOperationBase	PosterizeRotateSaturate	SharpnessShearXShearYSolarizeSolarizeAdd
TranslateX
TranslateYmin_magmax_magreturnc                 C   :   | | krt tj d|  d| dtddd|fddS )	zReturn ShearX op.< is a symmetric operation that `- min_mag == max_mag`. Got [, ]N      ?        Tmagnitude_rangesymmetric_megnitude)
ValueErrorr   __name__r   r    r!   ]/home/ubuntu/.local/lib/python3.10/site-packages/kornia/augmentation/auto/rand_augment/ops.pyshear_x(   
   
r#   c                 C   r   )	zReturn ShearY op.r   r   r   Nr   r   Tr   )r   r   r   r    r!   r!   r"   shear_y1   r$   r%   c                 C   r   )	zReturn TranslateX op.r   r   r   Nr   r   Tr   )r   r   r   r    r!   r!   r"   translate_x:   r$   r&   c                 C   r   )	zReturn TranslateY op.r   r   r   Nr   r   Tr   )r   r   r   r    r!   r!   r"   translate_yC   r$   r'   c                 C   r   )	zReturn rotate op.r   r   r   Nr   r   Tr   )r   r	   r   r    r!   r!   r"   rotateL   r$   r(   c                 C      t dS )zReturn AutoConstrast op.r   )r   r    r!   r!   r"   auto_contrastU      r*   c                 C   r)   )zReturn invert op.r   )r   r    r!   r!   r"   invertZ   r+   r,   c                 C   r)   )zReturn equalize op.r   )r   r    r!   r!   r"   equalize_   r+   r-   c                 C      t dd| |fdS )zReturn solarize op.Nr   r   )r   r    r!   r!   r"   solarized      r0   c                 C   r.   )zReturn SolarizeAdd op.Nr   r/   )r   r    r!   r!   r"   solarize_addi   r1   r2   c                 C   r.   )zReturn posterize op.Nr   r/   )r   r    r!   r!   r"   	posterizen   r1   r3   c                 C   r.   )zReturn contrast op.Nr   r/   )r   r    r!   r!   r"   contrasts   r1   r4   c                 C   r.   )zReturn brightness op.Nr   r/   )r   r    r!   r!   r"   
brightnessx   r1   r5   c                 C   r.   )zReturn sharpness op.Nr   r/   )r   r    r!   r!   r"   	sharpness}   r1   r6   c                 C   r.   )zReturn color op.Nr   r/   )r
   r    r!   r!   r"   color   r1   r7   N)"__doc__#kornia.augmentation.auto.operationsr   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   floatr#   r%   r&   r'   r(   r*   r,   r-   r0   r2   r3   r4   r5   r6   r7   r!   r!   r!   r"   <module>   s"   H					