GiZ?e5dGdde Z@e5dGdde ZAe@ZBy))Iterable) lru_cache)AnyOptionalUnionN) BatchFeature)BaseImageProcessor) center_crop)convert_to_rgbdivide_to_patchesget_resize_output_image_sizeget_size_with_aspect_ratiogroup_images_by_shapereorder_images) normalize)rescale)resize) ChannelDimension ImageInput ImageTypeSizeDictget_image_size#get_image_size_for_max_height_widthget_image_typeget_max_height_widthinfer_channel_dimension_formatis_valid_imageload_image_as_tensor) ImagesKwargsUnpack) TensorTypeis_torch_availableis_torchvision_availableis_vision_availablelogging)is_rocm_platformis_torchdynamo_compilingrequires)PILImageResampling) functional)pil_torch_interpolation_mappingtorch_pil_interpolation_mapping)torch torchvision)backendsc FeZdZdZdeeffd ZedefdZ ede fdZ de e e ze e e zfdZ d9d ed ed zd e ezd zd eddeeddf dZd edefdZ d:de ddeded zde d zdeded zded zdeeddffdZ d;d ddedddeddf dZe d;d dd eeefd!ed"deddf d#Zd dd$eddfd%Zd dd&eeezd'eeezddfd(Zed)* dis_fastzTorchvisionBackend.is_fast]s  ` r@cy)B `str`: The backend used by this image processor. r/r6rHs r>backendzTorchvisionBackend.backendjs r@image_url_or_urlsct|ttfr|Dcgc]}|j|c}St|tr t |St |r|Stdt|cc}w)z Convert a single or a list of URLs / paths into `torch.Tensor` objects. Already-valid image objects (tensors, numpy arrays, PIL Images) are passed through unchanged so that callers who pre-load images are unaffected. z=only a single or a list of entries is supported but got type=) isinstancelisttuple fetch_imagesstrrr TypeErrortype)r<rMxs r>rRzTorchvisionBackend.fetch_imagesqsv '$ 72CDQD%%a(D D )3 /'(9: : - .$ $[\`ar\s[tuv v EsA5Nimagedo_convert_rgbinput_data_formatdevicez torch.device torch.Tensorc ft|}|tjtjtjfvrt d||r|j |}|tjk(rtj|}n6|tjk(r#tj|j}|jdk(r|jd}| t|}|tj k(r!|j#dddj}||j%|}|S)z/Process a single image for torchvision backend.Unsupported input image type rr)rrPILTORCHNUMPY ValueErrorr tvF pil_to_tensorr. from_numpy contiguousndim unsqueezerrLASTpermuteto)r<rWrXrYrZr3 image_types r> process_imagez TorchvisionBackend.process_images$E* immY__iooN Nu E   0 5 5 5MM!Q*557E  HHV$E r@ct|SzConvert an image to RGB format.r r<rWs r>r z!TorchvisionBackend.convert_to_rgb e$$r@imagespad_size fill_value padding_mode return_maskdisable_grouping is_nested)r[r[c |@|jr |jstd|d|j|jf}n t|}t |||\} } i} i} | j D]\} }|j dd}|d|dz }|d|dz }|dks|dkrtd|d |d||k7rdd||f}tj|||| }|| | <|sstj|tj d dddddf}d|d d|dd|df<|| | <t| | | }|rt| | | }||fS|S)z5Pad images using Torchvision with batched operations.NCPad size must contain 'height' and 'width' keys only. Got pad_size=.)rxryrrzrPadding dimensions are negative. Please make sure that the `pad_size` is larger than the image size. Got pad_size=z , image_size=)fillrvdtype.)ry) heightwidthrbrritemsshapercpadr. zeros_likeint64r)r<rsrtrurvrwrxryr3grouped_imagesgrouped_images_indexprocessed_images_groupedprocessed_masks_groupedrstacked_images image_sizepadding_height padding_widthpadding stacked_masksprocessed_imagesprocessed_maskss r>rzTorchvisionBackend.pads  OO #fgofppq!rss 8H+F3H/D %50 ,,$& "$%3%9%9%; ? !E>'--bc2J%a[:a=8N$QK*Q-7M!]Q%6 008zzlRSUX%a?!$z`l!m.< $U + % 0 0u{{ STWYZ\]_`T` a GH c?Z]?OjmOCD1>'.# ?&**BDXdmn ,-DFZfopO#_4 4r@sizeresamplez7PILImageResampling | tvF.InterpolationMode | int | None antialiasc |#t|ttfr t|}n|}ntj j }|tj jk(r/tjdtj j}|jr?|jr3t|jdd|j|j}n|jr(t||jdt j"}n|j$r?|j&r3t)|jdd|j$|j&}n@|j*r%|j,r|j*|j,f}nt/d|dt1rt3r|j5||||St j6||||S) z"Resize an image using Torchvision.Na You have used a torchvision backend image processor with LANCZOS resample which not yet supported for torch.Tensor. BICUBIC resample will be used as an alternative. Please fall back to a pil backend image processor if you want full consistency with the original model.r}Frdefault_to_squarerYjSize must contain 'height' and 'width' keys, or 'max_height' and 'max_width', or 'shortest_edge' key. Got r| interpolationr)rOr*intr,rcInterpolationModeBILINEARLANCZOSrFrGBICUBIC shortest_edge longest_edgerrrrFIRST max_height max_widthrrrrbr(r'_compile_friendly_resizer)r<rWrrrr3rnew_sizes r>rzTorchvisionBackend.resizes  (%7$=> ? I ( 11::M C1199 9   A   1199M   $"3"31 RS!""!!H   3''"'"2"8"8 H __:5::<;Ldoo_c_m_mnH [[TZZ TZZ0H6  $ %*:*<00-QZ[ [zz%R[\\r@rrztvF.InterpolationModec|jtjk(r|jdz }t j ||||}|dz}tj |dkDd|}tj |dkd|}|jjtj}|St j ||||}|S)zOA wrapper around tvF.resize for torch.compile compatibility with uint8 tensors.rr) rr.uint8floatrcrwhereroundrk)rWrrrs r>rz+TorchvisionBackend._compile_friendly_resizes ;;%++ %KKMC'EJJuhmW`aECKEKK S%8EKK 1e4EKKM$$U[[1E JJuhmW`aE r@scalec ||zS)z5Rescale an image by a scale factor using Torchvision.r6r<rWrr3s r>rzTorchvisionBackend.rescale"su}r@meanstdc 0tj|||S)z%Normalize an image using Torchvision.)rcrr<rWrrr3s r>rzTorchvisionBackend.normalize+s}}UD#..r@ )maxsize do_normalize image_mean image_std do_rescalerescale_factorc|r>|r!_fuse_mean_std_and_rescale_factorz4TorchvisionBackend._fuse_mean_std_and_rescale_factor5sO ,j@C.DXYJ Yv>#BVWIJ9j00r@c|j||||||j\}}}|r3|j|jtj ||}|S|r|j ||}|S)zFRescale and normalize images using Torchvision (fused for efficiency).)rrrrrrZr)rrZrrkr.float32r)r<rsrrrrrs r>rescale_and_normalizez(TorchvisionBackend.rescale_and_normalizeFs-1,R,R%!!)== -S- ) Iz ^^FIIEMMI$BJPYZF \\&.9F r@c 4|j |jtd|j|jdd\}}|j|j}}||kDs||kDrv||kDr||z dznd||kDr||z dznd||kDr ||z dzdznd||kDr ||z dzdzndg}t j ||d}|jdd\}}||k(r||k(r|St||z dz } t||z dz } t j|| | ||S) z'Center crop an image using Torchvision.N=The size dictionary must have keys 'height' and 'width'. Got r}r^rr)r~g@) rrrbkeysrrcrrcrop) r<rWrr3 image_height image_width crop_height crop_width padding_ltrbcrop_top crop_lefts r>r zTorchvisionBackend.center_crop_sS ;; $**"4\]a]f]f]h\ijk k$)KK$4! k"&++tzzZ  #{\'A3= 3Kk)a/QR5@<5O|+1UV7AK7Ok)A-!3UV9D|9S|+a/A5YZ L GGE _preprocesszTorchvisionBackend._preprocess{s&*0EV^n/o,,!#%3%9%9%; ; !E>!%>W_!`,: "5 ) ;((>@TU0E^fv/w,,#% %3%9%9%; = !E>!%!1!1.)!L!77 NL*V_N/= $U + =**BDXY #xx(88^nxo .2B!CQ_``r@)NNN)NrconstantFFF)NT)NNNNNN)'__name__ __module__ __qualname____doc__r!r r9propertyboolrIrSrLrPrRrrrrmr rrrrQrr staticmethodrrrrrrrrr r"r r __classcell__r=s@r>r2r2Us|K' !5'    wcDIoT#Y.Ow&'+;?+/ !!t !!11D8 ! ( ! & ! !F%J%:%"!"#-!(-!&0 ^$0 0 $J 0 Dj 0  0 +0 $;0  u34nD E0 lOS 3]3]3]L 3]  3] 3]j<@ S/ 78   $  //huo%/Xe_ $ /  /r%)1504"&'++/1Tk1DK'$.14;&- 1 4K 1  1(1 11     DK' 4;& 2MMM  M8-a^$-a-a -a L -a  -a-a-a-a-aDK'$.-a4;&--at -aT/-a+-a j(4/!-a$ %-ar@r2)visionc:eZdZdZdeeffd ZedefdZ ede fdZ d*de d edzd e e zdzdeedejf d Zde de fd Z d+d eejdededzde dzdedeeejeejfeejzf dZ d*dejdedddedzdejf dZdejdedejfdZdejdeeezdeeezdejfdZdejdedejfdZd eejdededdded ed!ed"ed#ed$eeezdzd%eeezdzd&edzdedzd'e ezdzdefd(Zde e e!fffd) Z"xZ#S), PilBackendz9PIL/NumPy backend for portable CPU-only image processing.r3c Ht|di||jdi|yr5r7r;s r>r9zPilBackend.__init__r?r@rAc.tjdy)rCrDFrErHs r>rIzPilBackend.is_fasts  ` r@cy)rKpilr6rHs r>rLzPilBackend.backends r@NrWrXrYc t|}|tjtjtjfvrt d||r|j |}|tjk(r9tj|}|jdk\r8|tjn|}n#|tjk(r|j}|jdk(rtj|d}| t|}|tjk(r0t|tj rtj"|d}|S)z'Process a single image for PIL backend.r]r^r)axis)r^rr)rrr_r`rarbr nparrayrgrrinumpy expand_dimsrrOndarray transpose)r<rWrXrYr3rls r>rmzPilBackend.process_images$E* immY__iooN Nu E   0 5 5 5%, UI6 r@ct|Srorprqs r>r zPilBackend.convert_to_rgbrrr@rsrtrurvrwc |@|jr |jstd|d|j|j}}nt|\}}g} g} |D]} t | t j \} } || z }|| z }|dks|dkrtd|d|d| d| d | |k7s| |k7r@d d|fd|ff}|d k(rtj| |d | } ntj| || } | j| |stj||ftj}d|d| d| f<| j||r| | fS| S)z)Pad images to specified size using NumPy.Nr{r| channel_dimrzsPadding dimensions are negative. Please make sure that the `pad_size` is larger than the image size. Got pad_size=(z, z), image_size=(z).)rrr)modeconstant_values)rrr) rrrbrrrrrrappendzerosr)r<rsrtrurvrwr3 target_height target_widthrrrWrrrr pad_widthmasks r>rzPilBackend.pads  OO #fgofppq!rss*2//8>>v*F 'M< -E*5>N>T>TUMFE*V3N(50M!]Q%6 11>r,_e^ffhinhooqs &%<*?$a%81m:LM :-FF5)*V`aEFF5),GE  # #E *xx =RXXN()WfWfuf_%&&t,3 -6 #_4 4r@rrzPILImageResampling | None reducing_gapc T|>t|ttfs(t|tvr t|}ntj}||ntj}|j rN|j rBt|tj\}}t||f|j |j }n|j r(t||j dtj}n|jrN|jrBt|tj\}}t||f|j|j}n@|jr%|j r|j|j f}nt#d|dt%||||tjtjS)z Resize an image using PIL/NumPy.rFrrr|)rrr data_formatrY)rOr*rr-rrrrrrrrrrrrrrb np_resize) r<rWrrrr3rrrs r>rzPilBackend.resize"su   8>PRU=V(W.:xKj?j:8D-66'389K9T9T   $"3"3*5>N>T>TUMFE1""!!H   3''"'"2"8"8 H __*5>N>T>TUMFE:FE?DOO]a]k]klH [[TZZ TZZ0H6   %(...44   r@rc Xt||tjtjS)z/Rescale an image by a scale factor using NumPy.)rrrY) np_rescalerrrs r>rzPilBackend.rescaleUs) (...44   r@rrc Zt|||tjtjS)zNormalize an image using NumPy.)rrrrY) np_normalizerrrs r>rzPilBackend.normalizecs, (...44   r@c |j |jtd|jt ||j|jft j t j S)z!Center crop an image using NumPy.r)rrrY)rrrbrnp_center_croprr)r<rWrr3s r>r zPilBackend.center_cropssg ;; $**"4\]a]f]f]h\ijk k ++tzz*(...44   r@rrrrrrrrrrc $g}|D]f}|r|j|||}|r|j||}|r|j||}| r|j|| | }|j |h| r|j || }t d|i|S)z2Preprocess using PIL backend (portable, CPU-only).r)rtrr)rr rrrrr )r<rsrrrrrrrrrrrrtrr3rrWs r>rzPilBackend._preprocesss& +E %dX N(( : UN;uj)D  # #E * + #xx(88xL .2B!CQ_``r@c|t|}|jddjdr |ddd|d<|S)Nimage_processor_typePil)r8to_dictgetendswith)r<processor_dictr=s r>rzPilBackend.to_dictsK*   4b 9 B B5 I5CDZ5[\_]_5`N1 2r@)NN)NrrF)$rrrrr!r r9rrrIrSrLrrrrrmr rPrrrQrrrrrrr r"r rdictrrrrs@r>rrsC' !5'   '+;? ""t "!11D8 " & "  "H%J%:%"!"#-! 1 RZZ 1 1 $J 1 Dj 1  1  tBJJbjj!11 2T"**5E E1 n15#' 1 zz1 1 . 1 Dj 1  1 f  zz       zz huo% Xe_ $     zz    ""aRZZ "a"a "a . "a  "a"a"a"a"aDK'$."a4;&-"at "aT/"aj(4/"a" #"aHc3hr@r)Ccollections.abcr functoolsrtypingrrrrrimage_processing_baser image_processing_utilsr image_transformsr r r r rrrrrr rrrr image_utilsrrrrrrrrrrrprocessing_utilsr r!utilsr"r#r$r%r&utils.import_utilsr'r(r)r*r.torchvision.transforms.v2r+rcr,r- get_loggerrrFr2rBaseImageProcessorFastr6r@r>r%s%''/6    3UT/;]]&*#&*#   H % +,Ra+Ra-Raj  ;A#A AJ,r@