cCi"SrSSKrSSKJrJr SSKrSSKJr SSK J r J r J r SSK JrJr SSKJrJrJrJrJrJr SS KJr SS KJrJr S S KJr "S S\ 5r\"SS\55rS/r g)z%Fast Image processor class for OWLv2.N)OptionalUnion) functional)BaseImageProcessorFast BatchFeatureDefaultFastImageProcessorKwargs)group_images_by_shapereorder_images)OPENAI_CLIP_MEANOPENAI_CLIP_STDChannelDimension ImageInputPILImageResamplingSizeDict)Unpack) TensorTypeauto_docstring)OwlViTImageProcessorFastc\rSrSrSrg)Owlv2FastImageProcessorKwargs-N)__name__ __module__ __qualname____firstlineno____static_attributes__ra/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/owlv2/modular_owlv2.pyrr-scr rc\rSrSr\R r\r\ r SSS.r Sr Sr SrSrSr\rSrSrS\\4Sjr\S \S\\4S j5rS!S S S \S S 4SjjrS!S \S S\\S \S \S 4SjjrS"SS S\ S\S S 4Sjjr!S \S S\S\ S\SS\S\S\S\S\\"\\\4S\\"\\\4S\\S\\"\#\$4S \%4Sjr&S r'g)#Owlv2ImageProcessorFast0i)heightwidthgp?TNkwargsc 2[R"U40UD6 gN)r__init__)selfr's r!r* Owlv2ImageProcessorFast.__init__?s''77r imagesc 0[R"X40UD6$r))r preprocess)r+r-r's r!r/"Owlv2ImageProcessorFast.preprocessBs%00HHHr z torch.Tensorconstant_valuereturncURSSup4[X45nXS- nXT- nSSXv4n[R"XUS9n U $)z, Pad an image with zeros to the given size. Nr)fill)shapemaxFpad) r+r-r1r%r&size pad_bottom pad_rightpadding padded_images r! _pad_images#Owlv2ImageProcessorFast._pad_imagesFsP RS) 6!] L a/uuV>B r disable_groupingc [XS9upV0nUR5HupURU US9n XU'M [Xv5n U $)zx Unlike the Base class `self.pad` where all images are padded to the maximum image size, Owlv2 pads an image to square. rA)r1)r itemsr?r ) r+r-rAr1r'grouped_imagesgrouped_images_indexprocessed_images_groupedr6stacked_imagesprocessed_imagess r!r9Owlv2ImageProcessorFast.padSsg0EV/o,#% %3%9%9%; !E!---.N/=U + &<**BYr imager: anti_aliasingc URUR4nURn[R"USS5R UR 5[R"U5R UR 5- nU(Ga#UcUS- S- RSS9nO[R"U5[R"U5-n[R"US:5(a [S5e[R"US:US:*-5(a[R"S5 [R"US:H5(aUn O[S[R"SU-5R5-S-n [ R""XSU S4UR%5S 9n OUn [ R&"XRUR4S S 9n U $) a Resize an image as per the original implementation. Args: image (`Tensor`): Image to resize. size (`dict[str, int]`): Dictionary containing the height and width to resize the image to. anti_aliasing (`bool`, *optional*, defaults to `True`): Whether to apply anti-aliasing when downsampling the image. anti_aliasing_sigma (`float`, *optional*, defaults to `None`): Standard deviation for Gaussian kernel when downsampling the image. If `None`, it will be calculated automatically. rNr)minzFAnti-aliasing standard deviation must be greater than or equal to zerozWAnti-aliasing standard deviation greater than zero but not down-sampling along all axesr)sigmaF)r: antialias)r%r&r6torchtensortodeviceclamp atleast_1d ones_likeany ValueErrorwarningswarnceilintr8 gaussian_blurtolistresize) r+rKr:rLanti_aliasing_sigmar' output_shape input_shapefactorsfiltered kernel_sizesouts r!raOwlv2ImageProcessorFast.resizeks, TZZ0 kk ,,{12/225<<@5<!77 E:N/= $U + &<**BDXY #xx(8_oxp /D /?0 ,"$%3%9%9%; !E>y $ ("/&6&<&< !,! 1>&u-&<((>@TU0E^fv/w,#% %3%9%9%; !E>!77~ZN/= $U + &<**BDXYCQ5;;'7Q?Wg.2B!CQ_``r r)rt)TN)(rrrrrBILINEARresampler rpr rqr:rnrjrmrorlr valid_kwargs crop_sizedo_center_croprr*rrr/floatr?listrboolr9rrarstrrrrrrr r!r#r#0s!**H!JIC (DNIJL F0LIN8(E!F8IIv>[7\II . % R` "!$  ^$ #4.   n   8# 666 6 6p>a^$>a>a >a  56 >a  >a>a>a>aU5$u+#567>aE%e"456>a#4.>a!sJ!78>a >ar r#)!__doc__r[typingrrrRtorchvision.transforms.v2rr8image_processing_utils_fastrrr image_transformsr r image_utilsr r rrrrprocessing_utilsrutilsrr#owlvit.image_processing_owlvit_fastrrr#__all__rr r!rsy," 5 F'KJ$CIpa6papaf % %r