rhk#dZddlZddlmZmZddlmZddlmZm Z ddl m Z m Z ddl mZmZmZmZmZmZdd lmZdd lmZmZmZmZmZerddlZerdd lmZn erdd l mZGd d e Z!eGddeZ"dgZ#y)z%Fast Image processor class for OWLv2.N)OptionalUnion)OwlViTImageProcessorFast) BatchFeatureDefaultFastImageProcessorKwargs)group_images_by_shapereorder_images)OPENAI_CLIP_MEANOPENAI_CLIP_STDChannelDimension ImageInputPILImageResamplingSizeDict)Unpack) TensorTypeauto_docstringis_torch_availableis_torchvision_availableis_torchvision_v2_available) 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__z/var/www/html/ai-insurance-compliance-backend/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 8tjdi|y)Nr!)r__init__)selfr(s r#r*z Owlv2ImageProcessorFast.__init__Ps+ "++5f5r"imagesc 8tj|fi|S)N)r preprocess)r+r,r(s r#r.z"Owlv2ImageProcessorFast.preprocessSs4')44VFvFFr"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,r/r&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)Nr>)r/)r itemsr=r ) r+r,r>r/grouped_imagesgrouped_images_indexprocessed_images_groupedr4stacked_imagesprocessed_imagess r#r7zOwlv2ImageProcessorFast.paddsz 0EV^n/o,,#% %3%9%9%; = !E>!---.N/= $U +  =**BDXYr"imager8 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. Nr)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)r8 antialias)r&r'r4torchtensortodeviceclamp atleast_1d ones_likeany ValueErrorwarningswarnceilintr6 gaussian_blurtolistresize) r+rGr8rHanti_aliasing_sigmar( output_shape input_shapefactorsfiltered kernel_sizesouts r#r^zOwlv2ImageProcessorFast.resizews, TZZ0 kk ,,{12/225<<@5<rmr(rBrCrDr4rErFresized_images_grouped resized_stackresized_imagess r# _preprocessz#Owlv2ImageProcessorFast._preprocesss"0EV^n/o,,#% %3%9%9%; = !E>!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 rkr rlr8rirfrhrjrr valid_kwargs crop_sizedo_center_croprr*rrr.floatr=listrrr7rr^rstrrrrzr!r"r#r%r%As!**H!JIC (DNIJL F0LIN6(E!F6GGv>[7\GG . % 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%)$rrXtypingrr7transformers.models.owlvit.image_processing_owlvit_fastrimage_processing_utils_fastrrimage_transformsr r image_utilsr r r rrrprocessing_utilsrutilsrrrrrrOtorchvision.transforms.v2rr6torchvision.transformsrr%__all__r!r"r#rs,"\F' 96$Cka6kaka\ % %r"