qiL4dZddlZddlmZmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZddlmZmZmZej4eZGdd eZd gZy) zImage processor class for Pvt.N)BaseImageProcessor BatchFeature get_size_dict)resizeto_channel_dimension_format) IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDChannelDimension ImageInputPILImageResamplinginfer_channel_dimension_formatis_scaled_imagemake_flat_list_of_imagesto_numpy_array valid_imagesvalidate_preprocess_arguments) TensorTypefilter_out_non_signature_kwargsloggingceZdZdZdgZddej dddddfdedee e fdzded ed e e zd ed e e e zdzd e e e zdzddffd Z ej ddfdejdee e fdede ezdzde ezdzdejf dZedddddddddej&df dededzdee e fdzdedzd edzd e dzd edzd e e e zdzd e e e zdzde ezdzde ezde ezdzfdZxZS)PvtImageProcessora Constructs a PVT image processor. Args: do_resize (`bool`, *optional*, defaults to `True`): Whether to resize the image's (height, width) dimensions to the specified `(size["height"], size["width"])`. Can be overridden by the `do_resize` parameter in the `preprocess` method. size (`dict`, *optional*, defaults to `{"height": 224, "width": 224}`): Size of the output image after resizing. Can be overridden by the `size` parameter in the `preprocess` method. resample (`PILImageResampling`, *optional*, defaults to `PILImageResampling.BICUBIC`): Resampling filter to use if resizing the image. Can be overridden by the `resample` parameter 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 the `do_rescale` parameter 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 the `rescale_factor` parameter in the `preprocess` method. do_normalize (`bool`, *optional*, defaults to `True`): Whether to normalize the image. Can be overridden by the `do_normalize` parameter in the `preprocess` method. image_mean (`float` or `list[float]`, *optional*, defaults to `IMAGENET_DEFAULT_MEAN`): 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 `IMAGENET_DEFAULT_STD`): 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. pixel_valuesTNgp? do_resizesizeresample do_rescalerescale_factor do_normalize image_mean image_stdreturnc  t |di| ||nddd}t|}||_||_||_||_||_||_||nt|_ |||_ yt|_ y)N)heightwidth) super__init__rrrrrrrr r r r!) selfrrrrrrr r!kwargs __class__s ^/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/pvt/image_processing_pvt.pyr)zPvtImageProcessor.__init__Hs~ "6"'tc-JT""$(   ,(2(>*DY&/&;AUimage data_formatinput_data_formatc t|}d|vsd|vrtd|j|d|df}t|f||||d|S)a Resize an image to `(size["height"], size["width"])`. Args: image (`np.ndarray`): Image to resize. size (`dict[str, int]`): Dictionary in the format `{"height": int, "width": int}` specifying the size of the output image. resample (`PILImageResampling`, *optional*, defaults to `PILImageResampling.BICUBIC`): `PILImageResampling` filter to use when resizing the image e.g. `PILImageResampling.BICUBIC`. data_format (`ChannelDimension` or `str`, *optional*): The channel dimension format for the output image. If unset, the channel dimension format of the input image is used. 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. 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. Returns: `np.ndarray`: The resized image. r%r&zFThe `size` dictionary must contain the keys `height` and `width`. Got )rrr0r1)r ValueErrorkeysr)r*r/rrr0r1r+ output_sizes r-rzPvtImageProcessor.resize`syHT" 4 7$#6efjfofofqerst tH~tG}5   #/     r.imagesreturn_tensorsc z||n |j}||n |j}||n |j}||n |j}||n |j}||n |j }| | n |j } ||n |j}t|} t|}t|s tdt||||| ||||Dcgc] }t|}}|r#t|drtj!d| t#|d} |r"|Dcgc]}|j%|| || }}|r!|Dcgc]}|j'||| }}|r"|Dcgc]}|j)||| | }}|Dcgc]}t+|| | }}d |i}t-|| Scc}wcc}wcc}wcc}wcc}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`): Dictionary in the format `{"height": h, "width": w}` specifying the size of the output image after resizing. resample (`PILImageResampling` filter, *optional*, defaults to `self.resample`): `PILImageResampling` filter to use if resizing the image e.g. `PILImageResampling.BICUBIC`. 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 values between [0 - 1]. 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 if `do_normalize` is set to `True`. image_std (`float` or `list[float]`, *optional*, defaults to `self.image_std`): Image standard deviation to use 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.PYTORCH` or `'pt'`: Return a batch of type `torch.Tensor`. - `TensorType.NUMPY` or `'np'`: Return a batch of type `np.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. zSInvalid image type. Must be of type PIL.Image.Image, numpy.ndarray, or torch.Tensor)rrrr r!rrrrzIt 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/rrr1)r/scaler1)r/meanstdr1)input_channel_dimr)data tensor_type)rrrrrr r!rrrrr3rrrlogger warning_oncerrrescale normalizerr)r*r6rrrrrrr r!r7r0r1 size_dictr/r=s r- preprocesszPvtImageProcessor.preprocessst"+!6IDNN #-#9Zt '3'?|TEVEV '38+9+E4K^K^#-#9Zt !*!6IDNN 'tTYY!$' )&1F#rs s%!)%! 6<vay I  $ %i(^o pF $ 5Rc dF $U^opF ou ej '{N_ `  '>BBG=    sF$F)7F.F3<F8)__name__ __module__ __qualname____doc__model_input_namesr BICUBICbooldictstrintfloatlistr)npndarrayr rrFIRSTr rrD __classcell__)r,s@r-rr'si<((&*'9'A'A&-!1504VV38nt#V% V  V e VVDK'$.V4;&-V V8(:'A'A59;? / zz/ 38n/ % / ++d2 / !11D8 /  / b%&"&&*.2"&'+$(150426.>.D.D;?wCwC$;wC38nt# wC %t+ wC 4K wC wCTkwCDK'$.wC4;&-wCj(4/wC++wC!11D8wC'wCr.r)rHnumpyrQimage_processing_utilsrrrimage_transformsrr image_utilsr r r r r rrrrrrutilsrrr get_loggerrEr?r__all__r'r.r-r\sc%UUC    JI   H %bC*bCJ  r.