cCi;$SrSSKJrJrJr SSKJr SSKJrJ r SSK J r J r SSK JrJrJrJrJrJr SSKJrJr /S Qr/S QrS r"S S \5rS /rg)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?cURS:XaU$URS5n[R"SURS5n[R "X!5nURS5nU$)NRGBRGBA)rr)modeconvertrnewsizealpha_composite)image image_rgba backgroundrs n/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/idefics/image_processing_idefics.pyconvert_to_rgbr!&s_ zzU v&J6:??ODJ++JCO%--e4O c\^\rSrSrSrS/rSS\S\\\ \ \ 4S\\\ \ \ 4S \\S \ S \\\ 4S S4U4S jjjr SSSSSSS\ R4S\S \\S\\\\4S\\\ \ \ 4S\\\ \ \ 4S\\S \\ S \\ S\\\\ 4S \ 4SjjrSrU=r$)IdeficsImageProcessor3a 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_valuesNr image_size image_mean image_stdimage_num_channels do_rescalerescale_factorreturnc >[TU]"S0UD6 XlX@lUbUO[UlUbUO[ UlXPlX`l g)N) super__init__r'r*IDEFICS_STANDARD_MEANr(IDEFICS_STANDARD_STDr)r+r,) selfr'r(r)r*r+r,kwargs __class__s r r1IdeficsImageProcessor.__init__NsI "6"$"4(2(>*DY&/&;AU$,r"images transformreturn_tensorsc UbUO URnUbUO URnUbUO URnUbUO URnUbUO URnUbUO UR nX34n [ U[5(a[U5S:Xa/$URU5n[U5n[U5(d [S5eUbF[5(d [S5eSSKn UV s/sH o"U 5PM nn U R!U5$UV s/sHn [#U 5PM nn UV s/sHn [%U 5PM nn UV s/sHn ['X[(R*S9PM nn UVs/sHoR-XS9PM nnUV s/sHoR/XUS9PM nn UV s/sHn [1U [2R45PM nn [7SU0U S 9SnU$s sn fs sn fs sn fs sn fs snfs sn fs sn f) 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 installed)resample)rscale)meanstdr&)data tensor_type)r'r*r(r)r+r, isinstancelistlen fetch_imagesrr ValueErrorr ImportErrortorchstackr!rr rBICUBICrescale normalizer r FIRSTr )r4r8r*r'r(r)r9r+r,r:r5rrHxrs r preprocess IdeficsImageProcessor.preprocessas\$.#9Zt 3E3Q/W[WnWn#-#9Zt !*!6IDNN #-#9Zt +9+E4K^K^' fd # #F q(8I""6*)&1F##:   %''!"RSS ,23FqilFF3;;v& &.44V.#V4-34V.#V4PVWPV1&+=+E+EFPVWOUVve,,U,AvVMSTV...CVTRXYRXQ-a1A1G1GHRXYNF#;XYgh 454WVTYs*,G&G+/G0 "G52G:G?/$H)r+r(r*r'r)r,)NNrTgp?)__name__ __module__ __qualname____firstlineno____doc__model_input_namesintrrfloatrCboolr1rPYTORCHr dictstrrrO__static_attributes__ __classcell__)r6s@r r$r$3s0((:>9=,-,3--U5$u+#567-E%e"456 - %SM -  -c5j)- --,-./3:>9=(,%)*.;E;M;M\\%SM\T#s(^, \ U5$u+#567 \ E%e"456 \H%\TN\!\!sJ!78\ \\r"r$N)rVtypingrrrPILrimage_processing_utilsrr image_transformsr r image_utilsr r rrrrutilsrrr2r3r!r$__all__r/r"r rgsU),,FC4<; J.JZ # #r"