^j (dZddlmZddlZddlZddlmZddl m Z ddl m Z ddl mZmZdd lmZmZmZmZmZmZdd lmZmZdd lmZmZmZ dd eej<d fdededede eeff dZ!GddedZ"eGdde Z#dgZ$y)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_divisorreturnctr-t|tjr|jdd\}}nt |t j\}}||}}|t||z }||kr|} ||z} n||z} |} t| | |kDr|t| | z }|| z} || z} t| dzt| dz} } | |z|z} | |z|z} | | fS)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 /var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/bridgetower/image_processing_bridgetower.pyget_resize_output_image_sizer.%s ; E$/$5$5bc$:! k$2;L\LbLb$c! k &hH s<5 5Ek! K' \)   :y!H,3z955Z' I%  S 013y33G J|+l:J\)L8I y  ceZdZUdZeed<y)BridgeTowerImageProcessorKwargsz 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____doc__r%__annotations__r/r-r1r1Gs r/r1F)totalc#BeZdZdZeZddgZejZ e Z e Z ddiZdZddiZdZdZdZdZdZdZd eeffd Z d"d d d edddedd f fd Z d"ded ded edddedededededeeezdzdeeezdzdedzdedzdedzd ee zdzdede!f"d!Z"xZ#S)#BridgeTowerImageProcessorzGTorchvision backend for BridgeTower with custom resize and center_crop. pixel_values pixel_mask shortest_edgei FT kwargsc $t|di|y)Nr7)super__init__)selfr? __class__s r-rBz"BridgeTowerImageProcessor.__init__ds "6"r/imagersizeresamplez7PILImageResampling | tvF.InterpolationMode | int | Nonerrc |jstd|j|j}td|z}t ||||\}} t |d|t|| |d|S)z!Resize with size_divisor support.z@The `size` dictionary must contain the key `shortest_edge`. Got g)\?)rrrheightwidth)rErFrGr7)r= ValueErrorkeysr%r.rAresizer) rCrErFrGrr?rr output_height output_widthrDs r-rNz BridgeTowerImageProcessor.resizegs!!_`d`i`i`k_lmn n$$Z')*&B 76 ' # |w~ lC    r/images do_resizedo_center_crop crop_size do_rescalerescale_factor do_normalize image_meanN image_stddo_padpad_sizedisable_groupingreturn_tensorsc t||\}}i}|jD]!\}}|r|j||||}|||<#t||}t||\}}i}|jD]T\}}|r2|j |t |j |j }|j|||| | | }|||<Vt||}i}| r|j|d|\}}||d<||d<t|| S) z%Custom preprocessing for BridgeTower.)r\)rFrGrrI)rFT) return_maskr\r<r;)data tensor_type) ritemsrNr center_croprr=rescale_and_normalizepadr)rCrQrRrFrGrSrTrUrVrWrXrYrZr[r\r]rr?grouped_imagesgrouped_images_indexresized_images_groupedr!stacked_imagesresized_imagesprocessed_images_groupedprocessed_imagesr`processed_maskss r- _preprocessz%BridgeTowerImageProcessor._preprocess~se,0EV^n/o,,!#%3%9%9%; ; !E>!%^$QYht!u,: "5 ) ;((>@TU0E^fv/w,,#% %3%9%9%; = !E>!%!1!1"9P9PXaXoXo)p"2""77 NL*V_N/= $U + =**BDXY 04 dEU191 - o"1D /^>BBr/)r>)$r2r3r4r5r1 valid_kwargsmodel_input_namesr BICUBICrGr rXr rYrFdefault_to_squarerTrRrSrUrWrZrrrBrr%rNlistboolfloatstrrrrn __classcell__)rDs@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& '7Cr/r:)i i5r>)%r5typingrnumpynprtorchvision.transforms.v2rtvFimage_processing_backendsrimage_processing_utilsrimage_transformsrr image_utilsr r r r rrprocessing_utilsrrutilsrrrndarrayr%tupler.r1r:__all__r7r/r-rs- 7;2E5CC  !rzz>12! ! ! !  38_ !Dl%dC 2dCdCN ' 'r/