`i;$dZddlmZmZmZddlmZddlmZm Z ddl m Z m Z ddl mZmZmZmZmZmZddlmZmZgd Zgd Zd ZGd d eZd gZdS)z"Image processor class for Idefics.)CallableOptionalUnion)Image)BaseImageProcessor BatchFeature)resizeto_channel_dimension_format)ChannelDimension ImageInputPILImageResamplingmake_flat_list_of_imagesto_numpy_array valid_images) TensorTypeis_torch_available)g3<4'?gwg M?gy{ ?)gB91?gwt.?g U?c|jdkr|S|d}tjd|jd}tj||}|d}|S)NRGBRGBA)rr)modeconvertrnewsizealpha_composite)image image_rgba backgroundrs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/models/idefics/image_processing_idefics.pyconvert_to_rgbr!&sg zU v&&J6:?ODDJ+J CCO%--e44O ceZdZdZdgZ dded eeee efd eeee efd eed e d eeefddffd Z ddddddde j fded eedeeeefd eeee efd eeee efdeed ee d eedeeee fde fdZxZS)IdeficsImageProcessora Constructs a Idefics image processor. Args: image_size (`int`, *optional*, defaults to 224): Resize to image size image_mean (`float` or `list[float]`, *optional*, defaults to `IDEFICS_STANDARD_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. Can be overridden by the `image_mean` parameter in the `preprocess` method. image_std (`float` or `list[float]`, *optional*, defaults to `IDEFICS_STANDARD_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. Can be overridden by the `image_std` parameter in the `preprocess` method. image_num_channels (`int`, *optional*, defaults to 3): Number of image channels. 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. pixel_valuesNrTp? image_size image_mean image_stdimage_num_channels do_rescalerescale_factorreturnc tjdi|||_||_||nt|_||nt |_||_||_ dS)N) super__init__r(r+IDEFICS_STANDARD_MEANr)IDEFICS_STANDARD_STDr*r,r-) selfr(r)r*r+r,r-kwargs __class__s r r2zIdeficsImageProcessor.__init__Nsh ""6"""$"4(2(>**DY&/&;AU$,r"images transformreturn_tensorsc   ||nj}||nj}njnj||nj}nj||f t |trt|dkrgS |}t|}t|stdDtstdddl} fd|D}| |Sd|D}d|D} fd|D}fd |D}fd |D}d |D}t#d |i| d }|S)a| Preprocess a batch of images. Args: images (`ImageInput`): A list of images to preprocess. image_size (`int`, *optional*, defaults to `self.image_size`): Resize to image size image_num_channels (`int`, *optional*, defaults to `self.image_num_channels`): Number of image channels. image_mean (`float` or `list[float]`, *optional*, defaults to `IDEFICS_STANDARD_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. Can be overridden by the `image_mean` parameter in the `preprocess` method. image_std (`float` or `list[float]`, *optional*, defaults to `IDEFICS_STANDARD_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. Can be overridden by the `image_std` parameter in the `preprocess` method. transform (`Callable`, *optional*, defaults to `None`): A custom transform function that accepts a single image can be passed for training. For example, `torchvision.Compose` can be used to compose multiple transforms. If `None` - an inference mode is assumed - and then a preset of inference-specific transforms will be applied to the images 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. Returns: a PyTorch tensor of the processed images NrzkInvalid image type. Must be of type PIL.Image.Image, numpy.ndarray, torch.Tensor, tf.Tensor or jax.ndarray.z.To pass in `transform` torch must be installedc&g|] }|Sr0r0).0xr9s r z4IdeficsImageProcessor.preprocess..s!333qiill333r"c,g|]}t|Sr0)r!r=r>s r r?z4IdeficsImageProcessor.preprocess.. 444.##444r"c,g|]}t|Sr0)rrAs r r?z4IdeficsImageProcessor.preprocess..rBr"cHg|]}t|tjS))resample)r rBICUBIC)r=r>rs r r?z4IdeficsImageProcessor.preprocess..s,WWW1&D+=+EFFFWWWr"c>g|]}|S))rscale)rescale)r=rr-r5s r r?z4IdeficsImageProcessor.preprocess..s)VVVe$,,U.,AAVVVr"c@g|]}|S))meanstd) normalize)r=r>r)r*r5s r r?z4IdeficsImageProcessor.preprocess..s+TTT$...CCTTTr"cBg|]}t|tjSr0)r r FIRSTrAs r r?z4IdeficsImageProcessor.preprocess..s'YYYQ-a1A1GHHYYYr"r%)data tensor_type)r(r+r)r*r,r- isinstancelistlen fetch_imagesrr ValueErrorr ImportErrortorchstackr ) r5r8r+r(r)r*r9r,r-r:r6rXrs ` ``` ` @r preprocessz IdeficsImageProcessor.preprocessas\$.#9ZZt 3E3Q//W[Wn#-#9ZZt !*!6IIDN #-#9ZZt +9+E4K^J' fd # # F q(8(8I""6**)&11F## :   %'' T!"RSSS LLL3333F333F;;v&& &54V44444V444WWWWPVWWWVVVVVvVVVTTTTTTVTTTYYRXYYYNF#;XXXYgh r")r&NNrTr')__name__ __module__ __qualname____doc__model_input_namesintrrfloatrSboolr2rPYTORCHr dictstrrrZ __classcell__)r7s@r r$r$3s0((:>9=,-,3---U5$u+#567-E%e"456 - %SM -  -c5j)- ------,-./3:>9=(,%)*.;E;M\\\%SM\T#s(^, \ U5$u+#567 \ E%e"456 \H%\TN\!\!sJ!78\ \\\\\\\\r"r$N)r^typingrrrPILrimage_processing_utilsrr image_transformsr r image_utilsr r rrrrutilsrrr3r4r!r$__all__r0r"r rnsB)(,,,,,,,,,,FFFFFFFFCCCCCCCC43333333<;;;;;   JJJJJ.JJJZ # #r"