Ë
    T^(h÷.  ã                   óê   — d Z ddlmZmZmZmZ ddlmZmZm	Z	 ddl
mZmZmZ ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZmZ  ej>                  e «      Z! e«       rddl"Z" G d	„ d
e«      Z#d
gZ$y)z!Image processor class for SigLIP.é    )ÚDictÚListÚOptionalÚUnioné   )ÚBaseImageProcessorÚBatchFeatureÚget_size_dict)Úconvert_to_rgbÚresizeÚto_channel_dimension_format)ÚIMAGENET_STANDARD_MEANÚIMAGENET_STANDARD_STDÚChannelDimensionÚ
ImageInputÚPILImageResamplingÚinfer_channel_dimension_formatÚis_scaled_imageÚmake_flat_list_of_imagesÚto_numpy_arrayÚvalid_imagesÚvalidate_preprocess_arguments)Ú
TensorTypeÚfilter_out_non_signature_kwargsÚis_vision_availableÚloggingNc                   óä  ‡ — e Zd ZdZdgZddej                  ddddddf	dedee	e
f   ded	ed
ee
ef   dedeeeee   f      deeeee   f      dee   ddfˆ fd„Z e«       dddddddddej$                  ddfdedee   dee	e
f   ded	ee   d
ee   dee   deeeee   f      deeeee   f      deee	ef      dee   deee	ef      dee   dej,                  j,                  fd„«       Zˆ xZS )ÚSiglipImageProcessora€  
    Constructs a SigLIP image processor.

    Args:
        do_resize (`bool`, *optional*, defaults to `True`):
            Whether to resize the image's (height, width) dimensions to the specified `size`. Can be overridden by
            `do_resize` in the `preprocess` method.
        size (`Dict[str, int]` *optional*, defaults to `{"height": 224, "width": 224}`):
            Size of the image after resizing. Can be overridden by `size` in the `preprocess` method.
        resample (`PILImageResampling`, *optional*, defaults to `Resampling.BICUBIC`):
            Resampling filter to use if resizing the image. Can be overridden by `resample` in the `preprocess` method.
        do_rescale (`bool`, *optional*, defaults to `True`):
            Whether to rescale the image by the specified scale `rescale_factor`. Can be overridden by `do_rescale` in
            the `preprocess` method.
        rescale_factor (`int` or `float`, *optional*, defaults to `1/255`):
            Scale factor to use if rescaling the image. Can be overridden by `rescale_factor` in the `preprocess`
            method.
        do_normalize (`bool`, *optional*, defaults to `True`):
            Whether to normalize the image by the specified mean and standard deviation. Can be overridden by
            `do_normalize` in the `preprocess` method.
        image_mean (`float` or `List[float]`, *optional*, defaults to `[0.5, 0.5, 0.5]`):
            Mean to use if normalizing the image. This is a float or list of floats the length of the number of
            channels in the image. Can be overridden by the `image_mean` parameter in the `preprocess` method.
        image_std (`float` or `List[float]`, *optional*, defaults to `[0.5, 0.5, 0.5]`):
            Standard deviation to use if normalizing the image. This is a float or list of floats the length of the
            number of channels in the image. Can be overridden by the `image_std` parameter in the `preprocess` method.
            Can be overridden by the `image_std` parameter in the `preprocess` method.
        do_convert_rgb (`bool`, *optional*, defaults to `True`):
            Whether to convert the image to RGB.
    Úpixel_valuesTNgp?Ú	do_resizeÚsizeÚresampleÚ
do_rescaleÚrescale_factorÚdo_normalizeÚ
image_meanÚ	image_stdÚdo_convert_rgbÚreturnc
                 óÜ   •— t        ‰| �  di |
¤Ž |�|ndddœ}|�|nt        }|�|nt        }|| _        || _        || _        || _        || _        || _	        || _
        || _        |	| _        y )Néà   )ÚheightÚwidth© )ÚsuperÚ__init__r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   )Úselfr    r!   r"   r#   r$   r%   r&   r'   r(   ÚkwargsÚ	__class__s              €úp/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/models/siglip/image_processing_siglip.pyr0   zSiglipImageProcessor.__init__R   s‚   ø€ ô 	‰ÑÑ"˜6Ò"ØÐ'‰t¸ÀcÑ-JˆØ#-Ð#9‘ZÔ?Uˆ
Ø!*Ð!6‘IÔ<Qˆ	à"ˆŒØˆŒ	Ø ˆŒØ$ˆŒØ,ˆÔØ(ˆÔØ$ˆŒØ"ˆŒØ,ˆÕó    ÚimagesÚreturn_tensorsÚdata_formatÚinput_data_formatc           
      óê  — |�|n| j                   }|�|n| j                  }t        |dd¬«      }|�|n| j                  }|�|n| j                  }|�|n| j
                  }|�|n| j                  }|�|n| j                  }|	�|	n| j                  }	|�|n| j                  }t        |«      }t        |«      st        d«      ‚t        |||||	|||¬«       |r|D �cg c]  }t        |«      ‘Œ }}|D �cg c]  }t        |«      ‘Œ }}|r#t!        |d   «      rt"        j%                  d«       |€t'        |d   «      }|r(|d   |d	   }}|D �cg c]  }t)        |||f||¬
«      ‘Œ }}|r!|D �cg c]  }| j+                  |||¬«      ‘Œ }}|r"|D �cg c]  }| j-                  |||	|¬«      ‘Œ }}|D �cg c]  }t/        |||¬«      ‘Œ }}d|i}t1        ||
¬«      S c c}w c c}w c c}w c c}w c c}w c c}w )a:  
        Preprocess an image or batch of images.

        Args:
            images (`ImageInput`):
                Image to preprocess. Expects a single or batch of images with pixel values ranging from 0 to 255. If
                passing in images with pixel values between 0 and 1, set `do_rescale=False`.
            do_resize (`bool`, *optional*, defaults to `self.do_resize`):
                Whether to resize the image.
            size (`Dict[str, int]`, *optional*, defaults to `self.size`):
                Size of the image after resizing.
            resample (`int`, *optional*, defaults to `self.resample`):
                Resampling filter to use if resizing the image. This can be one of the enum `PILImageResampling`. Only
                has an effect if `do_resize` is set to `True`.
            do_rescale (`bool`, *optional*, defaults to `self.do_rescale`):
                Whether to rescale the image.
            rescale_factor (`float`, *optional*, defaults to `self.rescale_factor`):
                Rescale factor to rescale the image by if `do_rescale` is set to `True`.
            do_normalize (`bool`, *optional*, defaults to `self.do_normalize`):
                Whether to normalize the image.
            image_mean (`float` or `List[float]`, *optional*, defaults to `self.image_mean`):
                Image mean to use for normalization. Only has an effect if `do_normalize` is set to `True`.
            image_std (`float` or `List[float]`, *optional*, defaults to `self.image_std`):
                Image standard deviation to use for normalization. Only has an effect if `do_normalize` is set to
                `True`.
            return_tensors (`str` or `TensorType`, *optional*):
                The type of tensors to return. Can be one of:
                - Unset: Return a list of `np.ndarray`.
                - `TensorType.TENSORFLOW` or `'tf'`: Return a batch of type `tf.Tensor`.
                - `TensorType.PYTORCH` or `'pt'`: Return a batch of type `torch.Tensor`.
                - `TensorType.NUMPY` or `'np'`: Return a batch of type `np.ndarray`.
                - `TensorType.JAX` or `'jax'`: Return a batch of type `jax.numpy.ndarray`.
            data_format (`ChannelDimension` or `str`, *optional*, defaults to `ChannelDimension.FIRST`):
                The channel dimension format for the output image. Can be one of:
                - `"channels_first"` or `ChannelDimension.FIRST`: image in (num_channels, height, width) format.
                - `"channels_last"` or `ChannelDimension.LAST`: image in (height, width, num_channels) format.
                - Unset: Use the channel dimension format of the input image.
            input_data_format (`ChannelDimension` or `str`, *optional*):
                The channel dimension format for the input image. If unset, the channel dimension format is inferred
                from the input image. Can be one of:
                - `"channels_first"` or `ChannelDimension.FIRST`: image in (num_channels, height, width) format.
                - `"channels_last"` or `ChannelDimension.LAST`: image in (height, width, num_channels) format.
                - `"none"` or `ChannelDimension.NONE`: image in (height, width) format.
            do_convert_rgb (`bool`, *optional*, defaults to `self.do_convert_rgb`):
                Whether to convert the image to RGB.
        r!   F)Ú
param_nameÚdefault_to_squarezkInvalid image type. Must be of type PIL.Image.Image, numpy.ndarray, torch.Tensor, tf.Tensor or jax.ndarray.)r#   r$   r%   r&   r'   r    r!   r"   r   z­It looks like you are trying to rescale already rescaled images. If the input images have pixel values between 0 and 1, set `do_rescale=False` to avoid rescaling them again.r,   r-   )Úimager!   r"   r9   )r=   Úscaler9   )r=   ÚmeanÚstdr9   )Úinput_channel_dimr   )ÚdataÚtensor_type)r    r!   r
   r"   r#   r$   r%   r&   r'   r(   r   r   Ú
ValueErrorr   r   r   r   ÚloggerÚwarning_oncer   r   ÚrescaleÚ	normalizer   r	   )r1   r6   r    r!   r"   r#   r$   r%   r&   r'   r7   r8   r9   r(   r=   r,   r-   rB   s                     r4   Ú
preprocesszSiglipImageProcessor.preprocessn   sn  € ð~ "+Ð!6‘I¸D¿N¹Nˆ	ØÐ'‰t¨T¯Y©YˆÜ˜T¨fÈÔNˆØ'Ð3‘8¸¿¹ˆØ#-Ð#9‘Z¸t¿¹ˆ
Ø+9Ð+E™È4×K^ÑK^ˆØ'3Ð'?‘|ÀT×EVÑEVˆØ#-Ð#9‘Z¸t¿¹ˆ
Ø!*Ð!6‘I¸D¿N¹Nˆ	Ø+9Ð+E™È4×K^ÑK^ˆä)¨&Ó1ˆä˜FÔ#Üð:óð ô 	&Ø!Ø)Ø%Ø!ØØØØõ		
ñ Ø9?Ö@°”n UÕ+Ð@ˆFÐ@ð 6<Ö<¨E”. Õ'Ð<ˆÐ<áœ/¨&°©)Ô4Ü×Ñðsôð
 Ð$ä >¸vÀa¹yÓ IÐáØ  ™N¨D°©M�EˆFð $öàô ˜U¨&°%¨À8Ð_pÖqðˆFð ñ
 ð $öàð —‘ 5°ÐRc�ÕdðˆFð ñ
 ð $öàð —‘ U°ÀÐ^o�ÕpðˆFð ð ouö
ØejÔ'¨¨{ÐN_Ö`ð
ˆð 
ð  Ð'ˆÜ °>ÔBÐBùòO Aùò =ùòùòùòùò

s$   ÃGÃ1GÅ
G!Å*G&ÆG+Æ/G0)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úmodel_input_namesr   ÚBICUBICÚboolr   ÚstrÚintr   Úfloatr   r   r0   r   r   ÚFIRSTr   r   ÚPILÚImagerI   Ú__classcell__)r3   s   @r4   r   r   0   s3  ø„ ñð> (Ð(Ðð Ø#Ø'9×'AÑ'AØØ,3Ø!Ø:>Ø9=Ø)-ñ-àð-ð �3˜�8‰nð-ð %ð	-ð
 ð-ð ˜c 5˜jÑ)ð-ð ð-ð ˜U 5¨$¨u©+Ð#5Ñ6Ñ7ð-ð ˜E %¨¨e©Ð"4Ñ5Ñ6ð-ð ! ™ð-ð 
õ-ñ8 %Ó&ð %)Ø#Ø'+Ø%)Ø*.Ø'+Ø:>Ø9=Ø;?Ø2B×2HÑ2HØDHØ)-ñBCàðBCð ˜D‘>ðBCð �3˜�8‰nð	BCð
 %ðBCð ˜T‘NðBCð ! ™ðBCð ˜t‘nðBCð ˜U 5¨$¨u©+Ð#5Ñ6Ñ7ðBCð ˜E %¨¨e©Ð"4Ñ5Ñ6ðBCð !  s¨J Ñ!7Ñ8ðBCð Ð.Ñ/ðBCð $ E¨#Ð/?Ð*?Ñ$@ÑAðBCð ! ™ðBCð 
�‰�‰òBCó 'ôBCr5   r   )%rM   Útypingr   r   r   r   Úimage_processing_utilsr   r	   r
   Úimage_transformsr   r   r   Úimage_utilsr   r   r   r   r   r   r   r   r   r   r   Úutilsr   r   r   r   Ú
get_loggerrJ   rE   rU   r   Ú__all__r.   r5   r4   ú<module>r_      sx   ðñ (ç .Ó .ç UÑ U÷ñ ÷
÷ ÷ ñ ÷ _Ó ^ð 
ˆ×	Ñ	˜HÓ	%€ñ ÔÛôACÐ-ô ACðH "Ð
"�r5   