hx# dZddlZddlmZmZddlmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZddlmZdd lmZmZmZmZmZd d lmZerddlZerdd lmZ n erdd l!mZ Gd deZ"eGddeZ#dgZ$y)z%Fast Image processor class for OWLv2.N)OptionalUnion)BaseImageProcessorFast BatchFeatureDefaultFastImageProcessorKwargs)group_images_by_shapereorder_images)OPENAI_CLIP_MEANOPENAI_CLIP_STDChannelDimension ImageInputPILImageResamplingSizeDict)Unpack) TensorTypeauto_docstringis_torch_availableis_torchvision_availableis_torchvision_v2_available)OwlViTImageProcessorFast) functionalc"eZdZUdZeeed<y)Owlv2FastImageProcessorKwargsa do_pad (`bool`, *optional*, defaults to `True`): Controls whether to pad the image. Can be overridden by the `do_pad` parameter in the `preprocess` method. If `True`, padding will be applied to the bottom and right of the image with grey pixels. do_padN)__name__ __module__ __qualname____doc__rbool__annotations__e/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/transformers/models/owlv2/modular_owlv2.pyrr7s TNr$rcteZdZejZeZeZ dddZ dZ dZ dZ dZdZeZdZdZdeefdZededeefd Zddd d ed d fd Z dded deed ed ed fdZ d dd deded d fdZ ded dedededdededededee!eeefdee!eeefdeedee!e"e#fd e$fdZ%y)!Owlv2ImageProcessorFasti)heightwidthgp?TNkwargsc 0tj|fi|yN)r__init__)selfr*s r%r-z Owlv2ImageProcessorFast.__init__Ps''77r$imagesc 0tj||fi|Sr,)r preprocess)r.r/r*s r%r1z"Owlv2ImageProcessorFast.preprocessSs%00vHHHr$z torch.Tensorconstant_valuereturnc|jdd\}}t||}||z }||z }dd||f}tj|||} | S)z< Pad an image with zeros to the given size. Nr)fill)shapemaxFpad) r.r/r2r(r)size pad_bottom pad_rightpadding padded_images r% _pad_imagesz#Owlv2ImageProcessorFast._pad_imagesWs\ RS) 65!F] 5L aJ/uuVW>B r$disable_groupingct||\}}i}|jD]\}}|j||}|||<t||} | S)NrA)r2)r itemsr@r ) r.r/rAr2grouped_imagesgrouped_images_indexprocessed_images_groupedr7stacked_imagesprocessed_imagess r%r:zOwlv2ImageProcessorFast.paddsz 0EV^n/o,,#% %3%9%9%; = !E>!---.N/= $U +  =**BDXYr$imager; anti_aliasingc |j|jf}|j}tj|ddj |j tj|j |j z }|r||dz dz jd}ntj|tj|z}tj|dkr tdtj|dkD|dkzrtjdtj|dk(r|} n]dtjd|zjzdz} t!j"|| d| df|j% } n|} t!j&| |j|jfd } | S) az 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)r7torchtensortodeviceclamp atleast_1d ones_likeany ValueErrorwarningswarnceilintr9 gaussian_blurtolistresize) r.rJr;rKanti_aliasing_sigmar* output_shape input_shapefactorsfiltered kernel_sizesouts r%r`zOwlv2ImageProcessorFast.resizews, TZZ0 kk ,,{12/225<<@5<!77 NE:yN/= $U +  =**BDXY #xx(8K[x\ /D /?0 ,,"$%3%9%9%; > !E> $ ("/&6&<&< !,! 1>&u- >((>@TU0E^fv/w,,#% %3%9%9%; = !E>!77~|ZQZN/= $U +  =**BDXYCQ5;;'7Q?Wg.2B!CQ_``r$)g?)TN)&rrrrBILINEARresampler rmr rnr;rkrhrjrlrr valid_kwargs crop_sizedo_center_croprr-rrr1floatr@listrr!r:rr`rstrrrr|r#r$r%r'r'As!**H!JIC (DNIJL F0LIN8(E!F8IIv>[7\II . % R` "!$  ^$ #4.   n   .# 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')%r rZtypingrrimage_processing_utils_fastrrrimage_transformsr r image_utilsr r r rrrprocessing_utilsrutilsrrrrr#owlvit.image_processing_owlvit_fastrrQtorchvision.transforms.v2rr9torchvision.transformsrr'__all__r#r$r%rs," F'K 96$Cka6kaka\ % %r$