



"""simple docstring"""

import warnings

from ...processing_utils import ProcessorMixin
from ...tokenization_utils_base import BatchEncoding







class    __lowercase     (					_UpperCAmelCase):







						"""simple docstring"""







						_A		:							Union[str, Any]							       =		["""image_processor""", """tokenizer"""]
						_A		:							str							       =		"""ChineseCLIPImageProcessor"""
						_A		:							Tuple							       =		("""BertTokenizer""", """BertTokenizerFast""")







						def __init__(self   ,		lowercase__=None   ,		lowercase__=None   ,		**lowercase__       ):
											snake_case_					:		Any 					=					None
											if "feature_extractor" in kwargs:
																warnings.warn(
																    """The `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor`"""
																    """ instead."""   ,		lowercase__   ,		)
																snake_case_					:		Optional[Any] 					=					kwargs.pop("""feature_extractor"""       )

											snake_case_					:		str 					=					image_processor if image_processor is not None else feature_extractor
											if image_processor is None:
																raise ValueError("""You need to specify an `image_processor`."""       )
											if tokenizer is None:
																raise ValueError("""You need to specify a `tokenizer`."""       )

											super().__init__(lowercase__   ,		lowercase__       )
											snake_case_					:		Union[str, Any] 					=					self.image_processor







						def __call__(self   ,		lowercase__=None   ,		lowercase__=None   ,		lowercase__=None   ,		**lowercase__       ):

											if text is None and images is None:
																raise ValueError("""You have to specify either text or images. Both cannot be none."""       )

											if text is not None:
																snake_case_					:		Any 					=					self.tokenizer(lowercase__   ,		return_tensors=lowercase__   ,		**lowercase__       )

											if images is not None:
																snake_case_					:		Tuple 					=					self.image_processor(lowercase__   ,		return_tensors=lowercase__   ,		**lowercase__       )

											if text is not None and images is not None:
																snake_case_					:		List[Any] 					=					image_features.pixel_values
																return encoding
											elif text is not None:
																return encoding
											else:
																return BatchEncoding(data=dict(**lowercase__       )   ,		tensor_type=lowercase__       )







						def 						__UpperCamelCase							(self   ,		*lowercase__   ,		**lowercase__       ):
											return self.tokenizer.batch_decode(*lowercase__   ,		**lowercase__       )







						def 						__UpperCamelCase							(self   ,		*lowercase__   ,		**lowercase__       ):
											return self.tokenizer.decode(*lowercase__   ,		**lowercase__       )







						@property
						def 						__UpperCamelCase							(self       ):
											snake_case_					:		Optional[int] 					=					self.tokenizer.model_input_names
											snake_case_					:		Union[str, Any] 					=					self.image_processor.model_input_names
											return list(dict.fromkeys(tokenizer_input_names + image_processor_input_names       )       )







						@property
						def 						__UpperCamelCase							(self       ):
											warnings.warn(
											    """`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead."""   ,		lowercase__   ,		)
											return self.image_processor_class









"""simple docstring"""

from typing import Dict, List, Optional, Union

import numpy as np

from ...image_processing_utils import BaseImageProcessor, BatchFeature, get_size_dict
from ...image_transforms import (
    center_crop,
    get_resize_output_image_size,
    normalize,
    rescale,
    resize,
    to_channel_dimension_format,
)
from ...image_utils import (
    IMAGENET_DEFAULT_MEAN,
    IMAGENET_DEFAULT_STD,
    ChannelDimension,
    ImageInput,
    PILImageResampling,
    make_list_of_images,
    to_numpy_array,
    valid_images,
)
from ...utils import TensorType, is_vision_available, logging


if is_vision_available():
			import PIL


a_			   =					logging.get_logger(__name__)







class    __lowercase     (					_UpperCAmelCase):







						"""simple docstring"""







						_A		:							int							       =		["""pixel_values"""]







						def __init__(self   ,		lowercase__ = True   ,		lowercase__ = None   ,		lowercase__ = 0.9   ,		lowercase__ = PILImageResampling.BICUBIC   ,		lowercase__ = True   ,		lowercase__ = None   ,		lowercase__ = 1 / 2_55   ,		lowercase__ = True   ,		lowercase__ = True   ,		lowercase__ = None   ,		lowercase__ = None   ,		**lowercase__   ,		):
											super().__init__(**lowercase__       )
											snake_case_					:		Tuple 					=					size if size is not None else {"""shortest_edge""": 2_24}
											snake_case_					:		Union[str, Any] 					=					get_size_dict(lowercase__   ,		default_to_square=lowercase__       )
											snake_case_					:		str 					=					crop_size if crop_size is not None else {"""height""": 2_24, """width""": 2_24}
											snake_case_					:		Dict 					=					get_size_dict(lowercase__   ,		param_name="""crop_size"""       )

											snake_case_					:		Union[str, Any] 					=					do_resize
											snake_case_					:		List[str] 					=					size
											snake_case_					:		str 					=					crop_pct
											snake_case_					:		str 					=					resample
											snake_case_					:		Optional[Any] 					=					do_center_crop
											snake_case_					:		Dict 					=					crop_size
											snake_case_					:		int 					=					do_rescale
											snake_case_					:		Optional[int] 					=					rescale_factor
											snake_case_					:		str 					=					do_normalize
											snake_case_					:		str 					=					image_mean if image_mean is not None else IMAGENET_DEFAULT_MEAN
											snake_case_					:		List[str] 					=					image_std if image_std is not None else IMAGENET_DEFAULT_STD







						def 						__UpperCamelCase							(self   ,		lowercase__   ,		lowercase__   ,		lowercase__ = None   ,		lowercase__ = PILImageResampling.BICUBIC   ,		lowercase__ = None   ,		**lowercase__   ,		):
											snake_case_					:		Tuple 					=					get_size_dict(lowercase__   ,		default_to_square=lowercase__       )
											if "shortest_edge" not in size and ("height" not in size or "width" not in size):
																raise ValueError(f'size must contain \'height\' and \'width\' or \'shortest_edge\' as keys. Got {size.keys()}'       )
											if crop_pct is not None:
																if "shortest_edge" in size:
																					snake_case_					:		Optional[int] 					=					int(size["""shortest_edge"""] / crop_pct       )
																elif "height" in size and "width" in size:
																					if size["height"] == size["width"]:
																										snake_case_					:		Dict 					=					int(size["""height"""] / crop_pct       )
																					else:
																										snake_case_					:		List[str] 					=					(int(size["""height"""] / crop_pct       ), int(size["""width"""] / crop_pct       ))
																else:
																					raise ValueError("""Invalid size for resize: {}""".format(lowercase__       )       )

																snake_case_					:		List[Any] 					=					get_resize_output_image_size(lowercase__   ,		size=lowercase__   ,		default_to_square=lowercase__       )
											else:
																if "shortest_edge" in size:
																					snake_case_					:		Optional[int] 					=					get_resize_output_image_size(lowercase__   ,		size=size["""shortest_edge"""]   ,		default_to_square=lowercase__       )
																elif "height" in size and "width" in size:
																					snake_case_					:		int 					=					(size["""height"""], size["""width"""])
																else:
																					raise ValueError("""Invalid size for resize: {}""".format(lowercase__       )       )

											return resize(lowercase__   ,		size=lowercase__   ,		resample=lowercase__   ,		data_format=lowercase__   ,		**lowercase__       )







						def 						__UpperCamelCase							(self   ,		lowercase__   ,		lowercase__   ,		lowercase__ = None   ,		**lowercase__   ,		):
											snake_case_					:		int 					=					get_size_dict(lowercase__       )
											if "height" not in size or "width" not in size:
																raise ValueError(f'size must contain \'height\' and \'width\' as keys. Got {size.keys()}'       )
											return center_crop(lowercase__   ,		size=(size["""height"""], size["""width"""])   ,		data_format=lowercase__   ,		**lowercase__       )







						def 						__UpperCamelCase							(self   ,		lowercase__   ,		lowercase__   ,		lowercase__ = None   ,		**lowercase__   ,		):
											return rescale(lowercase__   ,		scale=lowercase__   ,		data_format=lowercase__   ,		**lowercase__       )







						def 						__UpperCamelCase							(self   ,		lowercase__   ,		lowercase__   ,		lowercase__   ,		lowercase__ = None   ,		**lowercase__   ,		):
											return normalize(lowercase__   ,		mean=lowercase__   ,		std=lowercase__   ,		data_format=lowercase__   ,		**lowercase__       )







						def 						__UpperCamelCase							(self   ,		lowercase__   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = None   ,		lowercase__ = ChannelDimension.FIRST   ,		**lowercase__   ,		):
											snake_case_					:		str 					=					do_resize if do_resize is not None else self.do_resize
											snake_case_					:		Any 					=					crop_pct if crop_pct is not None else self.crop_pct
											snake_case_					:		List[Any] 					=					resample if resample is not None else self.resample
											snake_case_					:		str 					=					do_center_crop if do_center_crop is not None else self.do_center_crop
											snake_case_					:		str 					=					do_rescale if do_rescale is not None else self.do_rescale
											snake_case_					:		Any 					=					rescale_factor if rescale_factor is not None else self.rescale_factor
											snake_case_					:		str 					=					do_normalize if do_normalize is not None else self.do_normalize
											snake_case_					:		List[Any] 					=					image_mean if image_mean is not None else self.image_mean
											snake_case_					:		int 					=					image_std if image_std is not None else self.image_std

											snake_case_					:		List[Any] 					=					size if size is not None else self.size
											snake_case_					:		Optional[Any] 					=					get_size_dict(lowercase__   ,		default_to_square=lowercase__       )
											snake_case_					:		List[Any] 					=					crop_size if crop_size is not None else self.crop_size
											snake_case_					:		int 					=					get_size_dict(lowercase__   ,		param_name="""crop_size"""       )

											snake_case_					:		List[str] 					=					make_list_of_images(lowercase__       )

											if not valid_images(lowercase__       ):
																raise ValueError(
																    """Invalid image type. Must be of type PIL.Image.Image, numpy.ndarray, """
																    """torch.Tensor, tf.Tensor or jax.ndarray."""       )

											if do_resize and size is None or resample is None:
																raise ValueError("""Size and resample must be specified if do_resize is True."""       )

											if do_center_crop and crop_pct is None:
																raise ValueError("""Crop_pct must be specified if do_center_crop is True."""       )

											if do_rescale and rescale_factor is None:
																raise ValueError("""Rescale factor must be specified if do_rescale is True."""       )

											if do_normalize and (image_mean is None or image_std is None):
																raise ValueError("""Image mean and std must be specified if do_normalize is True."""       )

											# All transformations expect numpy arrays.
											snake_case_					:		int 					=					[to_numpy_array(lowercase__       ) for image in images]

											if do_resize:
																snake_case_					:		str 					=					[self.resize(image=lowercase__   ,		size=lowercase__   ,		crop_pct=lowercase__   ,		resample=lowercase__       ) for image in images]

											if do_center_crop:
																snake_case_					:		Optional[int] 					=					[self.center_crop(image=lowercase__   ,		size=lowercase__       ) for image in images]

											if do_rescale:
																snake_case_					:		List[Any] 					=					[self.rescale(image=lowercase__   ,		scale=lowercase__       ) for image in images]

											if do_normalize:
																snake_case_					:		Optional[Any] 					=					[self.normalize(image=lowercase__   ,		mean=lowercase__   ,		std=lowercase__       ) for image in images]

											snake_case_					:		List[Any] 					=					[to_channel_dimension_format(lowercase__   ,		lowercase__       ) for image in images]

											snake_case_					:		Dict 					=					{"""pixel_values""": images}
											return BatchFeature(data=lowercase__   ,		tensor_type=lowercase__       )




