Ph"dZddlZddlmZmZddlZddlmZddl m Z m Z m Z ddl mZmZddlmZmZmZmZmZmZdd lmZdd lmZmZd d lmZGd de ZeGddeZdgZ y)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 eZdZy)Owlv2FastImageProcessorKwargsN)__name__ __module__ __qualname__a/var/www/html/saasai/venv/lib/python3.12/site-packages/transformers/models/owlv2/modular_owlv2.pyrr-srrcteZdZejZeZeZ 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 rr&z Owlv2ImageProcessorFast.__init__?s''77rimagesc 0tj||fi|Sr%)r preprocess)r'r(r#s rr*z"Owlv2ImageProcessorFast.preprocessBs%00vHHHrz 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(r+r!r"size pad_bottom pad_rightpadding padded_images r _pad_imagesz#Owlv2ImageProcessorFast._pad_imagesFs\ RS) 65!F] 5L aJ/uuVW>B rdisable_groupingc t||\}}i}|jD]\}} |j| |} | ||<t||} | S)z Unlike the Base class `self.pad` where all images are padded to the maximum image size, Owlv2 pads an image to square. r:)r+)r itemsr9r ) r'r(r:r+r#grouped_imagesgrouped_images_indexprocessed_images_groupedr0stacked_imagesprocessed_imagess rr3zOwlv2ImageProcessorFast.padSsu0EV^n/o,,#% %3%9%9%; !E>!---.N/= $U + &<**BDXYrimager4 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)r4 antialias)r!r"r0torchtensortodeviceclamp atleast_1d ones_likeany ValueErrorwarningswarnceilintr2 gaussian_blurtolistresize) r'rCr4rDanti_aliasing_sigmar# output_shape input_shapefactorsfiltered kernel_sizesouts rrYzOwlv2ImageProcessorFast.resizeks, TZZ0 kk ,,{12/225<<@5<r?r@r0rArBresized_images_grouped resized_stackresized_imagess r _preprocessz#Owlv2ImageProcessorFast._preprocesss"0EV^n/o,,#% %3%9%9%; !E>!77 NE:yN/= $U + &<**BDXY #xx(8_oxp /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)rk)TN)&rrrrBILINEARresampler rgr rhr4rerardrfrcr valid_kwargs crop_sizedo_center_croprr&rrr*floatr9listrboolr3rrYrstrrrrwrrrr 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 >arr )!__doc__rStypingrrrJtorchvision.transforms.v2rr2image_processing_utils_fastrrr image_transformsr r image_utilsr r rrrrprocessing_utilsrutilsrr#owlvit.image_processing_owlvit_fastrrr __all__rrrrsy," 5 F'KJ$CIpa6papaf % %r