qyi &SrSSKJr SSKrSSKrSSKJr SSK J r SSK J r SSK JrJr SS KJrJrJrJrJrJr SS KJrJr SS KJrJrJr SS \\R<S 4S\S\S\S\ \\44 Sjjr!"SS\SS9r"\"SS\ 55r#S/r$g)z&Image processor class for BridgeTower.)UnionN) functional)TorchvisionBackend) BatchFeature)group_images_by_shapereorder_images)OPENAI_CLIP_MEANOPENAI_CLIP_STDChannelDimensionPILImageResamplingSizeDictget_image_size) ImagesKwargsUnpack) TensorTypeauto_docstringis_torch_available input_image torch.Tensorshorterlonger size_divisorreturnc[5(a1[U[R5(aURSSupEO[ U[ RS9upEXpvU[XE5- nXE:aUn X-n OX-n Un [X5U:aU[X5- nX-n X-n [U S-5[U S-5pX-U-n X-U-n X4$)z7Get output image size after resizing with size_divisor.N) channel_dimg?) r isinstancetorchTensorshaperr FIRSTminmaxint) rrrr input_height input_widthmin_sizemax_sizescale new_height new_widths y/home/mande/repo/quber/.venv/lib/python3.13/site-packages/transformers/models/bridgetower/image_processing_bridgetower.pyget_resize_output_image_sizer.%s ; E E$/$5$5bc$:! k$2;L\LbLb$c!  h s<5 5E! ' )   :!H,3z55' %  S 013y33G +l:J)L8I   c$\rSrSr%Sr\\S'Srg)BridgeTowerImageProcessorKwargsGz size_divisor (`int`, *optional*, defaults to `self.size_divisor`): The size by which to make sure both the height and width can be divided. rN)__name__ __module__ __qualname____firstlineno____doc__r%__annotations____static_attributes__r3r/r-r1r1Gs r/r1F)totalc#X^\rSrSrSr\rSS/r\Rr \ r \ rSS0rSrSS0rSrSrSrSrSrS rS \\4U4S jjrS$S S S\SSS\SS 4 U4SjjjrS$S\S S\S\SSS\S\S\S\S\S\\\-S-S\\\-S-S\S-S\S-S \S-S!\ \!-S-S\S\"4"S"jjr#S#r$U=r%$)%BridgeTowerImageProcessorPzGTorchvision backend for BridgeTower with custom resize and center_crop. pixel_values pixel_mask shortest_edgei FT kwargsc &>[TU]"S0UD6 g)Nr3)super__init__)selfrC __class__s r-rF"BridgeTowerImageProcessor.__init__ds "6"r/imagersizeresamplez7PILImageResampling | tvF.InterpolationMode | int | Nonerrc >UR(d[SUR535eURn[SU-5n[ XXtS9up[ T U]"SU[XS9US.UD6$)z!Resize with size_divisor support.z@The `size` dictionary must contain the key `shortest_edge`. Got g)\?)rrrheightwidth)rJrKrLr3)rA ValueErrorkeysr%r.rEresizer) rGrJrKrLrrCrr output_height output_widthrHs r-rS BridgeTowerImageProcessor.resizegs!!_`d`i`i`k_lmn n$$Z')*&B 6' # w~ C    r/images do_resizedo_center_crop crop_size do_rescalerescale_factor do_normalize image_meanN image_stddo_padpad_sizedisable_groupingreturn_tensorsc [XS9unn0nUR5H#unnU(aURUX4US9nUUU'M% [UU5n[UUS9unn0nUR5HRunnU(a,UR U[ UR UR S9S9nURUXxXU 5nUUU'MT [UU5n0nU (aURUSUS9unnUUS'UUS'[UUS 9$) z%Custom preprocessing for BridgeTower.)rb)rKrLrrN)rKT) return_maskrbr@r?)data tensor_type) ritemsrSr center_croprrArescale_and_normalizepadr)rGrWrXrKrLrYrZr[r\r]r^r_r`rarbrcrrCgrouped_imagesgrouped_images_indexresized_images_groupedr!stacked_imagesresized_imagesprocessed_images_groupedprocessed_imagesrfprocessed_maskss r- _preprocess%BridgeTowerImageProcessor._preprocess~sQ,0EV/o,,!#%3%9%9%; !E>!%^$ht!u,: "5 )&<((>@TU0E^fv/w,,#% %3%9%9%; !E>!%!1!1"9P9PXaXoXo)p"2""77 LV_N/= $U +&<**BDXY 04 dEU191 - o"1D /^>BBr/r3)rB)&r4r5r6r7r8r1 valid_kwargsmodel_input_namesr BICUBICrLr r^r r_rKdefault_to_squarerZrXrYr[r]r`rrrFrr%rSlistboolfloatstrrrrtr: __classcell__)rHs@r-r=r=PsQ2L'6!))H!JI S !D #&IINJL FL#(G!H#    L       P#7C^$7C7C 7C L 7C  7C7C7C7C7CDK'$.7C4;&-7Ct 7CT/7C+7C j(4/!7C"#7C& '7C7Cr/r=)i i5rB)%r8typingrnumpynprtorchvision.transforms.v2rtvFimage_processing_backendsrimage_processing_utilsrimage_transformsrr image_utilsr r r r rrprocessing_utilsrrutilsrrrndarrayr%tupler.r1r=__all__r3r/r-rs- 7;2E5CC  !rzz>12! ! ! !  38_ !Dl%dC 2dCdCN ' 'r/