NgNddlZddlZddlZddlZddlmZmZmZmZddl Z ddl m Z ddl m cm Z ddl mZdZn #e$rdZYnwxYwddlmZddlZgdZGdd ZGd d ZGd d e jZGddZeedrKejjdejjdejjdejj dejj!dejj"diZ#n,ejdejdejdej dej!dej"diZ#de#$DZ%erKejdejdejdej dej!dej"diZ&de&$DZ'niZ(iZ&dZ)dZ*dZ+e*de*dfZ,d6dZ-GddZ. d7d!e j/d"ee0ee0fd#ee0ee0e0e0ffd$e1d%e j/f d&Z2Gd'd(e j3j4Z5 d7d!e j/d)e0d*e0d+e0d,e0d#ee0ee0e0e0ffd$e1d%e j/fd-Z6Gd.d/e j3j4Z7Gd0d1Z8Gd2d3Z9Gd4d5e j3j4Z:dS)8N)ListSequenceTupleUnion)InterpolationModeTF)Image)ToNumpyToTensorstr_to_interp_modestr_to_pil_interpinterp_mode_to_str!RandomResizedCropAndInterpolationCenterCropOrPadcenter_crop_or_pad crop_or_padRandomCropOrPad RandomPadResizeKeepRatio TrimBorder MaybeToTensorMaybePILToTensorceZdZdZdS)r ctj|tj}|jdkrtj|d}tj|d}|S)Ndtype)axis)nparrayuint8ndim expand_dimsrollaxis)selfpil_imgnp_imgs P/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/data/transforms.py__call__zToNumpy.__call__sM'222 ;??^F444FVQ'' N)__name__ __module__ __qualname__r*r+r)r r s#r+r c,eZdZdZejfdZdZdS)r z% ToTensor with no rescaling of valuesc||_dSNr)r&rs r)__init__zToTensor.__init__%s  r+c\tj||jS)Nr)F pil_to_tensortor)r&r's r)r*zToTensor.__call__(s%w''***<<r@s r)r3zMaybePILToTensor.__init__GrBr+cbt|tjr|Stj|S)z Note: A deep copy of the underlying array is performed. Args: pic (PIL Image): Image to be converted to tensor. Returns: Tensor: Converted image. )rDr9rEr5r6rGs r)r*zMaybePILToTensor.__call__Js- c5< ( ( Js###r+c |jjdSrJrKrLs r)rMzMaybePILToTensor.__repr__XrNr+rO) r,r-r.r8r3r*rPrMrQrRs@r)rrCsq $ $ $.#........r+r Resamplingnearestbilinearbicubicboxhamminglanczosci|]\}}|| Sr/r/.0abs r) rcrsPPPdaQPPPr+ci|]\}}|| Sr/r/r_s r)rcrc~s"X"X"XDAq1a"X"X"Xr+ct|Sr2)_str_to_pil_interpolationmode_strs r)r r s $X ..r+cDtr t|St|Sr2)has_interpolation_mode_str_to_torch_interpolationrfrgs r)r r s 3*844(22r+cDtr t|St|Sr2)rj_torch_interpolation_to_str_pil_interpolation_to_str)modes r)r r s /*400(..r+3Please provide only two dimensions (h, w) for size.c*t|tjrt|t|fSt|tr#t |dkr|d|dfSt |dkrt ||S)Nrr)rDnumbersNumberintrlen ValueError)size error_msgs r) _setup_sizerzs$''$4yy#d))##$!! c$ii1nnAwQ 4yyA~~### Kr+cBeZdZdZ d dZedZdZdZd S) raYCrop 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?rYcTt|ttfrt||_n ||f|_|d|dks|d|dkrt jd|dkr t |_nt||_||_ ||_ dS)Nrrrz"range should be of kind (min, max)random) rDlisttuplerxwarningswarn_RANDOM_INTERPOLATION interpolationr scaleratio)r&rxrrrs r)r3z*RandomResizedCropAndInterpolation.__init__s dT5M * * %d DIIt DI !HuQx  U1Xa%8%8 M> ? ? ? H $ $!6D  !3M!B!BD   r+c tj|\}}||z}tdD]}tj||z}t j|dt j|df}t jtj|} ttt j || z} ttt j || z } | |kr>| |kr8tj d|| z } tj d|| z } | | | | fcS||z }|t|kr/|} tt| t|z } nF|t|kr/|} tt| t|z} n|} |} || z dz} || z dz} | | | | 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. rrrr)r5get_image_sizerangeruniformmathlogexpruroundsqrtrandintminmax)imgrrimg_wimg_hareaattempt target_area log_ratio aspect_ratiotarget_wtarget_hijin_ratios r) get_paramsz,RandomResizedCropAndInterpolation.get_paramss',, uu}Ryy 0 0G .%047K%(++TXeAh-?-?@I8FNI$>??L5;+E!F!FGGHHH5;+E!F!FGGHHH5  X%6%6N1eh&677N1eh&677!Xx////5= c%jj H5CJJ!67788HH E " "H5CJJ!67788HHHH X ! # X ! #!Xx''r+c |||j|j\}}}}t|jt t frtj|j}n|j}tj ||||||j |S)z Args: img (PIL Image): Image to be cropped and resized. Returns: PIL Image: Randomly cropped and resized image. ) rrrrDrrrrchoicer5 resized_croprx)r&rrrhwrs r)r*z*RandomResizedCropAndInterpolation.__call__sx__S$*djAA 1a d(5$- 8 8 /"M$*<==MM .M~c1aAty-HHHr+ct|jttfr%dd|jD}nt |j}|jjd|j z}|dtd|j Dz }|dtd|j Dz }|d|z }|S) N c,g|]}t|Sr/r r`xs r) z>RandomResizedCropAndInterpolation.__repr__..!'Z'Z'Z!(:1(=(='Z'Z'Zr+ (size={0}z , scale={0}c36K|]}t|dVdSNr)r`ss r) z=RandomResizedCropAndInterpolation.__repr__.. *3T3TAE!QKK3T3T3T3T3T3Tr+z , ratio={0}c36K|]}t|dVdSrr)r`rs r)rz=RandomResizedCropAndInterpolation.__repr__.. rr+z, interpolation={0})) rDrrrjoinr rAr,formatrxrrr&interpolate_str format_strings r)rMz*RandomResizedCropAndInterpolation.__repr__s d(5$- 8 8 E!hh'Z'ZtGY'Z'Z'Z[[OO01CDDO/+2D2DTY2O2OO --e3T3T3T3T3T.T.TUUU --e3T3T3T3T3T.T.TUUU /66GGG r+N)r|r~rY r,r-r.r8r3 staticmethodrr*rMr/r+r)rrsz  "$$ *((((\((T I I I     r+rconstantr output_sizefill padding_moder<c.t|}|\}}tj|\}}}||ks||kr||kr||z dznd||kr||z dznd||kr ||z dzdznd||kr ||z dzdzndg} tj|| ||}tj|\}}}||kr||kr|St t ||z dz } t t ||z dz } tj|| | ||S)akCenter 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. rrrrrrg@)rzr5get_dimensionspadrurcrop) rrrr crop_height crop_width_ image_height image_width padding_ltrbcrop_top crop_lefts r)rrsl*k**K)K#$#3C#8#8 A|[K;#=#=/9K/G/GZ+ %! + +Q1<|1K1K[< 'A - -QR3= 3K3KZ+ % )a / /QR5@<5O5O[< '! + 1 1UV  eCD|LLL'('7'<'<$<  $ $ )D)DJ5,4;<<==HE;3s:;;<#>>>>>>>>r+rtopleftheightwidthc tj|\}}} ||z} ||z} |dks|dks | | ks| |krt| td| zdt| td| zdt| t| |z dt| t||z dg} tj|| ||}t|d}t|d}tj|||||S)z[ Crops and/or pads image to meet target size, with control over fill and padding_mode. rr)r5rrrrr) rrrrrrrrrrrightbottomrs r)rr_s$%#3C#8#8 A|[ 5LE 6\F axx377ek11Vl5J5J Au %q ) ) s1f~~%q ) ) K... 2 2 \3/// 3 3  eCD|LLL c1++C tQ<#>>>>>>>>r+rc2eZdZddZedZdZdS)rrc"||_||_dSr2) input_sizer)r&rrs r)r3zRandomPad.__init__s$ r+ctj|\}}t|d|z d}t|d|z d}tjd|}tjd|}||z }||z } |||| fS)Nrrr)r5rrrr) rrrrrrpad_leftpad_top pad_right pad_bottoms r)rzRandomPad.get_paramss(-- v*Q-%/33 :a=61155 >![11.L11(* !G+ )Z77r+cr|||j}tj|||j}|Sr2)rrr5rr)r&rpaddings r)r*zRandomPad.__call__s1//#t77eC$),, r+N)r)r,r-r.r3rrr*r/r+r)rrsR88\8r+rcVeZdZdZ d dZe dd Zd Zd Zd S)rz" Resize and Keep Aspect Ratio rYg333333?g?Fg?g(\?c 6t|ttfrt||_n ||f|_|dkr t|_nt ||_t||_||_ ||_ ||_ ||_ ||_ dS)z Args: size: longest: interpolation: random_scale_prob: random_scale_range: random_scale_area: random_aspect_prob: random_aspect_range: rN)rDrrrxrrr floatlongestrandom_scale_probrandom_scale_rangerandom_scale_arearandom_aspect_probrandom_aspect_range) r&rxrrrrrrrs r)r3zResizeKeepRatio.__init__s. dT5M * * %d DIIt DI H $ $!6D  !3M!B!BD W~~ !2"4!2"4#6   r+r}gHzG?ctj|ddx\}} } |\} } || z } | | z }t| ||zt| |d|z zz|dkrVt j|kr?t j|d|d}|rdt j|z }||f}nd}|dkrt j|kr~t j|dt j|df}t j t j|}t j|}|d|z |d|zf}fdt| |D}|S)zGet parameters rrNr}r)r}r}c@g|]\}}t||zz Sr/r)r`rfrs r)rz.ResizeKeepRatio.get_params..s-MMMAa!eem$$MMMr+) r5rrrrrrrrrzip)r target_sizerrrrrrrrimg_sizerrratio_hratio_w ratio_factor log_aspect aspect_factorrxrs @r)rzResizeKeepRatio.get_paramss#$"23"7"7";; ux((("("GW%%/#gw2G2G2PW<2XX q V]__7H%H%H!>*.(rr+rz, interpolation=z , longest=z.3fz, random_scale_prob=z, random_scale_range=(rz, rrrz, random_aspect_prob=z, random_aspect_range=(z)))rDrrrrr rAr,rrxrrrrrrs r)rMzResizeKeepRatio.__repr__&sk d(5$- 8 8 E!hh'Z'ZtGY'Z'Z'Z[[OO01CDDO/+2D2DTY2O2OO =O=== 8dl8888 L0FLLLL a 3A6`aa>B>VWX>Y`aaa a N1HNNNN c 4Q7bcc?C?WXY?Zbccc c r+N)rrYrrFrr)rrFrrrr/r+r)rrs $ +#! +$7$7$7$7L !*#! +%%%\%N(r+rc*eZdZdeffd ZdZxZS)r border_sizecVt||_dSr2)r?r3r)r&rrAs r)r3zTrimBorder.__init__9s) &r+ctj|\}}|jx}}t||}t||}t d|d|jzz }t d|d|jzz }tj|||||S)Nrr)r5rrrrr)r&rrrrrrrs r)rzTrimBorder.forward@s$$1%%d#qkk4||QA 00011Aq1t///00vc3fe444r+)r,r-r.rur3rrQrRs@r)rr7sT'''''''5555555r+r)rpr);rrsrrtypingrrrrr9torchvision.transforms transforms!torchvision.transforms.functional functionalr5rrj ImportErrorPILrnumpyr __all__r r rrhasattrrWNEARESTBILINEARBICUBICBOXHAMMINGLANCZOSrnitemsrfrmrk_pil_interpolation_to_torchr r r rrzrrErurPrnnModulerrrrrrr/r+r)r/sY  //////////// ++++++---------#CCCCCC!###"#   ========.....J'............8 75,  ) !:  ) e  )  ) !  y   y 5 y y !QP.G.M.M.O.OPPP %!9"J!9u!9!9 ##Y"X4O4U4U4W4W"X"X"X"$"$///333///,+J779K9KI9V9VW    ggggggggZ23& 'E'E \'E3S >*'ECsC}--.'E 'E  \ 'E'E'E'ET">">">">">eho">">">V23&11 \1 11 1  1 CsC}--. 11 \11118,>,>,>,>,>eho,>,>,>^,ttttttttn5555555555s:AA