a  d. @sddlZddlZddlZddlZddlmZmZddlZddlm m Z zddlm Z dZ WneytdZ Yn0ddlmZddlZGdddZGd d d Zeed rejjd ejjd ejjdejjdejjdejjdiZn(ejd ejd ejdejdejdejdiZddeDZe rbe jd e jd e jde jde jde jdiZ dde DZ!niZ"iZ ddZ#ddZ$ddZ%e$d e$dfZ&ddZ'GdddZ(d&ej)ee*ej)dd d!Z+Gd"d#d#ej,j-Z.Gd$d%d%Z/dS)'N)ListSequence)InterpolationModeTF)Imagec@seZdZddZdS)ToNumpycCs8tj|tjd}|jdkr(tj|dd}t|d}|SNdtype)Zaxis)nparrayuint8ndim expand_dimsrollaxisselfZpil_imgZnp_imgr]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/data/transforms.py__call__s   zToNumpy.__call__N)__name__ __module__ __qualname__rrrrrrsrc@s"eZdZejfddZddZdS)ToTensorcCs ||_dSNr)rr rrr__init__szToTensor.__init__cCsHtj|tjd}|jdkr(tj|dd}t|d}t|j|j dSr) r rrrrrtorchZ from_numpytor rrrrr!s   zToTensor.__call__N)rrrrfloat32rrrrrrrsr ResamplingZnearestbilinearZbicubicZboxZhammingZlanczoscCsi|]\}}||qSrr.0abrrr ?r'cCsi|]\}}||qSrrr#rrrr'Kr(cCst|Sr)_str_to_pil_interpolationZmode_strrrrstr_to_pil_interpQsr+cCstr t|St|SdSr)has_interpolation_mode_str_to_torch_interpolationr)r*rrrstr_to_interp_modeUsr.cCstr t|St|SdSr)r,_torch_interpolation_to_str_pil_interpolation_to_str)moderrrinterp_mode_to_str\sr2cCsZt|tjrt|t|fSt|trBt|dkrB|d|dfSt|dkrVt||S)Nrr ) isinstancenumbersNumberintrlen ValueError)size error_msgrrr _setup_sizefs  r<c@s6eZdZdZdddZeddZd d Zd d Zd S)!RandomResizedCropAndInterpolationaYCrop the given PIL Image to random size and aspect ratio with random interpolation. A crop of random size (default: of 0.08 to 1.0) of the original size and a random aspect ratio (default: of 3/4 to 4/3) of the original aspect ratio is made. This crop is finally resized to given size. This is popularly used to train the Inception networks. Args: size: expected output size of each edge scale: range of size of the origin size cropped ratio: range of aspect ratio of the origin aspect ratio cropped interpolation: Default: PIL.Image.BILINEAR g{Gz??g?gUUUUUU?r"cCsxt|ttfrt||_n ||f|_|d|dksD|d|dkrNtd|dkr^t|_n t||_||_ ||_ dS)Nrr3z"range should be of kind (min, max)random) r4listtupler:warningswarn_RANDOM_INTERPOLATION interpolationr.scaleratio)rr:rHrIrGrrrrs     z*RandomResizedCropAndInterpolation.__init__c Cs|jd|jd}tdD]}tj||}t|dt|df}ttj|}ttt ||}ttt ||} ||jdkr| |jdkrt d|jd| } t d|jd|} | | | |fSq|jd|jd} | t |kr"|jd}tt|t |} nB| t |krP|jd} tt| t |}n|jd}|jd} |jd| d} |jd|d} | | | |fS)aGet parameters for ``crop`` for a random sized crop. Args: img (PIL Image): Image to be cropped. scale (tuple): range of size of the origin size cropped ratio (tuple): range of aspect ratio of the origin aspect ratio cropped Returns: tuple: params (i, j, h, w) to be passed to ``crop`` for a random sized crop. rr3 r ) r:rangerAuniformmathlogexpr7roundsqrtrandintminmax) imgrHrIareaattemptZ target_areaZ log_ratioZ aspect_ratiowhijZin_ratiorrr get_paramss.      z,RandomResizedCropAndInterpolation.get_paramsc CsV|||j|j\}}}}t|jttfr8t|j}n|j}t ||||||j |S)z Args: img (PIL Image): Image to be cropped and resized. Returns: PIL Image: Randomly cropped and resized image. ) r\rHrIr4rGrCrBrAchoiceFZ resized_cropr:)rrUrZr[rYrXrGrrrrs z*RandomResizedCropAndInterpolation.__call__cCst|jttfr(ddd|jD}n t|j}|jjd|j }|dtdd|j D7}|dtd d|j D7}|d |7}|S) N cSsg|] }t|qSr)r2r$xrrr r(z>RandomResizedCropAndInterpolation.__repr__.. (size={0}z , scale={0}css|]}t|dVqdSNrP)r$srrr r(z=RandomResizedCropAndInterpolation.__repr__..z , ratio={0}css|]}t|dVqdSrdrf)r$rrrrrhr(z, interpolation={0})) r4rGrCrBjoinr2 __class__rformatr:rHrIrZinterpolate_str format_stringrrr__repr__s z*RandomResizedCropAndInterpolation.__repr__N)r>r@r" rrr__doc__r staticmethodr\rrorrrrr=ss  *r=)rU output_sizereturnc CsHt|tjrt|t|f}n*t|ttfrHt|dkrH|d|df}t|\}}}|\}}||ksr||kr||kr||dnd||kr||dnd||kr||ddnd||kr||ddndg}tj |||d}t|\}}}||kr||kr|Stt ||d} tt ||d} t || | ||S)ajCenter crops and/or pads the given image. If the image is torch Tensor, it is expected to have [..., H, W] shape, where ... means an arbitrary number of leading dimensions. If image size is smaller than output size along any edge, image is padded with 0 and then center cropped. Args: img (PIL Image or Tensor): Image to be cropped. output_size (sequence or int): (height, width) of the crop box. If int or sequence with single int, it is used for both directions. fill (int, Tuple[int]): Padding color Returns: PIL Image or Tensor: Cropped image. r3rr fillg@) r4r5r6r7rCrBr8r^Zget_dimensionspadrPZcrop) rUrsrv_Z image_heightZ image_widthZ crop_heightZ crop_widthZ padding_ltrbZcrop_topZ crop_leftrrrcenter_crop_or_pads& rycs8eZdZdZd fdd ZddZeddd ZZS) CenterCropOrPada/Crops the given image at the center. If the image is torch Tensor, it is expected to have [..., H, W] shape, where ... means an arbitrary number of leading dimensions. If image size is smaller than output size along any edge, image is padded with 0 and then center cropped. Args: size (sequence or int): Desired output size of the crop. If size is an int instead of sequence like (h, w), a square crop (size, size) is made. If provided a sequence of length 1, it will be interpreted as (size[0], size[0]). rcs"tt|dd|_||_dS)Nz3Please provide only two dimensions (h, w) for size.)r;)superrr<r:rv)rr:rvrkrrr s zCenterCropOrPad.__init__cCst||j|jdS)z Args: img (PIL Image or Tensor): Image to be cropped. Returns: PIL Image or Tensor: Cropped image. ru)ryr:rv)rrUrrrforwardszCenterCropOrPad.forward)rtcCs|jjd|jdS)Nz(size=))rkrr:)rrrrroszCenterCropOrPad.__repr__)r) rrrrqrr}strro __classcell__rrr|rrzs  rzc@s6eZdZdZdddZeddZd d Zd d Zd S)ResizeKeepRatioz Resize and Keep Ratio r"rcCsBt|ttfrt||_n ||f|_t||_t||_||_dSr) r4rBrCr:r.rGfloatlongestrv)rr:rrGrvrrrr#s     zResizeKeepRatio.__init__c sf|jddd}|\}}|\}}||}||} t|| |t|| d|fdd|D} | S)a Get parameters Args: img (PIL Image): Image to be cropped. target_size (Tuple[int, int]): Size of output Returns: tuple: params (h, w) and (l, r, t, b) to be passed to ``resize`` and ``pad`` respectively Nr r?csg|]}t|qSrrfr`rIrrrbBr(z.ResizeKeepRatio.get_params..)r:rTrS) rUZ target_sizer source_sizerYrXZtarget_hZtarget_wZratio_hZratio_wr:rrrr\2s  zResizeKeepRatio.get_paramscCs&|||j|j}t|||j}|S)z Args: img (PIL Image): Image to be cropped and resized. Returns: PIL Image: Resized, padded to at least target size, possibly cropped to exactly target size )r\r:rr^resizerG)rrUr:rrrrEszResizeKeepRatio.__call__cCsFt|j}|jjd|j}|d|d7}|d|jdd7}|S)Nrcz, interpolation=r~z , longest=z.3f)r2rGrkrrlr:rrmrrrroQs  zResizeKeepRatio.__repr__N)rr"rrprrrrrs   r)r)0rMr5rArDtypingrrrZ!torchvision.transforms.functionalZ transformsZ functionalr^rr, ImportErrorZPILrnumpyr rrhasattrr!ZNEARESTZBILINEARZBICUBICZBOXZHAMMINGZLANCZOSr0itemsr)r/r-Z_pil_interpolation_to_torchr+r.r2rFr<r=ZTensorr7rynnModulerzrrrrrsf         e(