U
|
©r_Ú ã @ sþ d Z ddlZddlZddlmZmZmZ dd Zdd Z dd Z
|
d
|
d Z d d Z dd Z dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Ze e eeeeeeeg Ze e eeeeeeeg Ze e eeeeeeeeeeeg ZdS )&z9
|
@author: liaoxingyu
|
@contact: sherlockliao01@gmail.com
|
é N)ÚImageÚImageOpsÚ ImageEnhancec C s t | tjrft| jdks t| jdkr>| dd
dd
df } t | d¡¡}t |tj
|
rb| ¡ S |S | j dkrt tj | tjdd¡}n| j dkr®t tj | tjdd¡}nb| j d krÒt tj | tjdd¡}n>| j d
|
krúd t tj | tjdd¡ }nt
|
tj | ¡ ¡¡}| j d kr"d }n | j dkr4d}n
|
t| j }| | jd | jd |¡}| dd¡ dd¡ ¡ }t |tj
|
r| ¡ S |S dS )zéConvert a ``PIL Image`` or ``numpy.ndarray`` to tensor.
|
|
See ``ToTensor`` for more details.
|
|
Args:
|
pic (PIL Image or numpy.ndarray): Image to be converted to tensor.
|
|
Returns:
|
Tensor: Converted image.
|
)é é r N)r r é ÚIF)ÚcopyzI;16ÚFÚ1éÿ ÚYCbCrr r r )Ú
|
isinstanceÚnpÚndarrayÚlenÚshapeÚAssertionErrorÚndimÚtorchÚ
|
from_numpyÚ transposeÚ
|
ByteTensorÚfloatÚmodeÚarrayÚint32Úint16Úfloat32Úuint8Ú ByteStorageÚ from_bufferÚtobytesÚviewÚsizeÚ
|
contiguous)ÚpicÚimgZnchannel© r( ú ./data/transforms/functional.pyÚ to_tensor s6
|
|
|
|
|
|
r* c C s t | | d S )aS Helper function to scale `val` between 0 and maxval .
|
Args:
|
level: Level of the operation that will be between [0, `PARAMETER_MAX`].
|
maxval: Maximum value that the operation can have. This will be scaled to
|
level/PARAMETER_MAX.
|
Returns:
|
An int that results from scaling `maxval` according to `level`.
|
é
|
)Úint©ÚlevelZmaxvalr( r( r) Ú int_parameter@ s r/ c C s t | | d S )aS Helper function to scale `val` between 0 and maxval.
|
Args:
|
level: Level of the operation that will be between [0, `PARAMETER_MAX`].
|
maxval: Maximum value that the operation can have. This will be scaled to
|
level/PARAMETER_MAX.
|
Returns:
|
A float that results from scaling `maxval` according to `level`.
|
g $@)r r- r( r( r) Úfloat_parameterL s r0 c C s t jjd| dS )Nç¹?)ÚlowÚhigh)r ÚrandomÚuniform)Únr( r( r) Ú sample_levelX s r7 c G s
|
t | ¡S ©N)r Ú autocontrast©Úpil_imgÚargsr( r( r) r9 \ s r9 c G s
|
t | ¡S r8 )r Úequalizer: r( r( r) r= ` s r= c G s t t|d}t | d| ¡S )Né )r/ r7 r Ú posterize©r; r. r< r( r( r) r? d s r? c G s2 t t|d}tj ¡ dkr"| }| j|tjdS )Né ç à?©Úresample)r/ r7 r r4 r5 Úrotater ÚBILINEAR)r; r. r< Údegreesr( r( r) rE i s rE c G s t t|d}t | d| ¡S )Né )r/ r7 r Úsolarizer@ r( r( r) rI p s rI c C sD t t|d}tj ¡ dkr"| }| j|tjd|ddddftjdS ©Ng333333Ó?rB r r rC © r0 r7 r r4 r5 Ú transformr ÚAFFINErF ©r; r. Ú
|
image_sizer( r( r) Úshear_xu s þrP c C sD t t|d}tj ¡ dkr"| }| j|tjddd|ddftjdS rJ rK rN r( r( r) Úshear_y~ s þrQ c C sL t t||d d }tj ¡ dkr*| }| j|tjdd|dddftjdS )Nr r rB r rC ©r/ r7 r r4 rL r rM rF rN r( r( r) Ú translate_x s þrS c C sL t t||d d }tj ¡ dkr*| }| j|tjddddd|ftjdS )Nr r rB r rC rR rN r( r( r) Ú translate_y s þrT c G s" t t|dd }t | ¡ |¡S ©NgÍÌÌÌÌÌü?r1 )r0 r7 r ÚColorÚenhancer@ r( r( r) Úcolor s rX c G s" t t|dd }t | ¡ |¡S rU )r0 r7 r ÚContrastrW r@ r( r( r) Úcontrast s rZ c G s" t t|dd }t | ¡ |¡S rU )r0 r7 r Ú
|
BrightnessrW r@ r( r( r) Ú
|
brightness¦ s r\ c G s" t t|dd }t | ¡ |¡S rU )r0 r7 r Ú SharpnessrW r@ r( r( r) Ú sharpness¬ s r^ ) Ú__doc__Únumpyr r ÚPILr r r r* r/ r0 r7 r9 r= r? rE rI rP rQ rS rT rX rZ r\ r^ Úaugmentations_reidZ augmentationsZaugmentations_allr( r( r( r) Ú<module> sl 4
|
þ þ þ
|
