o
    ϯih                     @   sH   d dl mZmZmZmZmZmZmZ dd Z		dde	de
fdd	Zd
S )    )	NormalizeComposeRandomResizedCropInterpolationModeToTensorResize
CenterCropc                 C   s
   |  dS )NRGB)convert)image r   T/home/ubuntu/.local/lib/python3.10/site-packages/laion_clap/clap_module/transform.py_convert_to_rgb   s   
r   g3<4'?gwgM?gy{ ?gB91?gwt.?g	U?
image_sizeis_trainc                 C   sP   t ||d}|rtt| dtjdtt |gS tt| tjdt| tt |gS )N)meanstd)g?g      ?)scaleinterpolation)r   )	r   r   r   r   BICUBICr   r   r   r   )r   r   r   r   	normalizer   r   r   image_transform	   s   r   N)r   r   )torchvision.transformsr   r   r   r   r   r   r   r   intboolr   r   r   r   r   <module>   s   $ 