





'''simple docstring'''



_SCREAMING_SNAKE_CASE   =      {
    '''Pillow''': '''Pillow<10.0.0''',
    '''accelerate''': '''accelerate>=0.20.3''',
    '''av''': '''av==9.2.0''',
    '''beautifulsoup4''': '''beautifulsoup4''',
    '''black''': '''black~=23.1''',
    '''codecarbon''': '''codecarbon==1.2.0''',
    '''cookiecutter''': '''cookiecutter==1.7.3''',
    '''dataclasses''': '''dataclasses''',
    '''datasets''': '''datasets!=2.5.0''',
    '''decord''': '''decord==0.6.0''',
    '''deepspeed''': '''deepspeed>=0.9.3''',
    '''diffusers''': '''diffusers''',
    '''dill''': '''dill<0.3.5''',
    '''evaluate''': '''evaluate>=0.2.0''',
    '''fairscale''': '''fairscale>0.3''',
    '''faiss-cpu''': '''faiss-cpu''',
    '''fastapi''': '''fastapi''',
    '''filelock''': '''filelock''',
    '''flax''': '''flax>=0.4.1,<=0.7.0''',
    '''ftfy''': '''ftfy''',
    '''fugashi''': '''fugashi>=1.0''',
    '''GitPython''': '''GitPython<3.1.19''',
    '''hf-doc-builder''': '''hf-doc-builder>=0.3.0''',
    '''huggingface-hub''': '''huggingface-hub>=0.14.1,<1.0''',
    '''importlib_metadata''': '''importlib_metadata''',
    '''ipadic''': '''ipadic>=1.0.0,<2.0''',
    '''isort''': '''isort>=5.5.4''',
    '''jax''': '''jax>=0.2.8,!=0.3.2,<=0.4.13''',
    '''jaxlib''': '''jaxlib>=0.1.65,<=0.4.13''',
    '''jieba''': '''jieba''',
    '''kenlm''': '''kenlm''',
    '''keras-nlp''': '''keras-nlp>=0.3.1''',
    '''librosa''': '''librosa''',
    '''nltk''': '''nltk''',
    '''natten''': '''natten>=0.14.6''',
    '''numpy''': '''numpy>=1.17''',
    '''onnxconverter-common''': '''onnxconverter-common''',
    '''onnxruntime-tools''': '''onnxruntime-tools>=1.4.2''',
    '''onnxruntime''': '''onnxruntime>=1.4.0''',
    '''opencv-python''': '''opencv-python''',
    '''optuna''': '''optuna''',
    '''optax''': '''optax>=0.0.8,<=0.1.4''',
    '''packaging''': '''packaging>=20.0''',
    '''parameterized''': '''parameterized''',
    '''phonemizer''': '''phonemizer''',
    '''protobuf''': '''protobuf''',
    '''psutil''': '''psutil''',
    '''pyyaml''': '''pyyaml>=5.1''',
    '''pydantic''': '''pydantic<2''',
    '''pytest''': '''pytest>=7.2.0''',
    '''pytest-timeout''': '''pytest-timeout''',
    '''pytest-xdist''': '''pytest-xdist''',
    '''python''': '''python>=3.8.0''',
    '''ray[tune]''': '''ray[tune]''',
    '''regex''': '''regex!=2019.12.17''',
    '''requests''': '''requests''',
    '''rhoknp''': '''rhoknp>=1.1.0,<1.3.1''',
    '''rjieba''': '''rjieba''',
    '''rouge-score''': '''rouge-score!=0.0.7,!=0.0.8,!=0.1,!=0.1.1''',
    '''ruff''': '''ruff>=0.0.241,<=0.0.259''',
    '''sacrebleu''': '''sacrebleu>=1.4.12,<2.0.0''',
    '''sacremoses''': '''sacremoses''',
    '''safetensors''': '''safetensors>=0.3.1''',
    '''sagemaker''': '''sagemaker>=2.31.0''',
    '''scikit-learn''': '''scikit-learn''',
    '''sentencepiece''': '''sentencepiece>=0.1.91,!=0.1.92''',
    '''sigopt''': '''sigopt''',
    '''starlette''': '''starlette''',
    '''sudachipy''': '''sudachipy>=0.6.6''',
    '''sudachidict_core''': '''sudachidict_core>=20220729''',
    '''tensorflow-cpu''': '''tensorflow-cpu>=2.6,<2.14''',
    '''tensorflow''': '''tensorflow>=2.6,<2.14''',
    '''tensorflow-text''': '''tensorflow-text<2.14''',
    '''tf2onnx''': '''tf2onnx''',
    '''timeout-decorator''': '''timeout-decorator''',
    '''timm''': '''timm''',
    '''tokenizers''': '''tokenizers>=0.11.1,!=0.11.3,<0.14''',
    '''torch''': '''torch>=1.9,!=1.12.0''',
    '''torchaudio''': '''torchaudio''',
    '''torchvision''': '''torchvision''',
    '''pyctcdecode''': '''pyctcdecode>=0.4.0''',
    '''tqdm''': '''tqdm>=4.27''',
    '''unidic''': '''unidic>=1.0.2''',
    '''unidic_lite''': '''unidic_lite>=1.0.7''',
    '''urllib3''': '''urllib3<2.0.0''',
    '''uvicorn''': '''uvicorn''',
}












'''simple docstring'''



from typing import Dict, List, Optional, Tuple, 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_STANDARD_MEAN,
    IMAGENET_STANDARD_STD,
    ChannelDimension,
    ImageInput,
    PILImageResampling,
    make_list_of_images,
    to_numpy_array,
    valid_images,
)
from ...utils import TensorType, is_torch_available, is_torch_tensor, logging


if is_torch_available():
	import torch


_SCREAMING_SNAKE_CASE   =      logging.get_logger(__name__)






class 							__lowercase    ( lowerCAmelCase__  ):







	'''simple docstring'''

	a      :			int     =    ["pixel_values"]


	def __init__(self       ,_lowerCamelCase = True       ,_lowerCamelCase = None       ,_lowerCamelCase = PILImageResampling.BILINEAR       ,_lowerCamelCase = True       ,_lowerCamelCase = None       ,_lowerCamelCase = True       ,_lowerCamelCase = 1 / 255       ,_lowerCamelCase = True       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,**_lowerCamelCase       ,)							->							None:







				'''simple docstring'''





				super().__init__(**_lowerCamelCase	)
				__lowercase			  =							size if size is not None else {'''shortest_edge''': 256}
				__lowercase			  =							get_size_dict(_lowerCamelCase       ,default_to_square=_lowerCamelCase	)
				__lowercase			  =							crop_size if crop_size is not None else {'''height''': 224, '''width''': 224}
				__lowercase			  =							get_size_dict(_lowerCamelCase       ,param_name='''crop_size'''	)
				__lowercase			  =							do_resize
				__lowercase			  =							size
				__lowercase			  =							resample
				__lowercase			  =							do_center_crop
				__lowercase			  =							crop_size
				__lowercase			  =							do_rescale
				__lowercase			  =							rescale_factor
				__lowercase			  =							do_normalize
				__lowercase			  =							image_mean if image_mean is not None else IMAGENET_STANDARD_MEAN
				__lowercase			  =							image_std if image_std is not None else IMAGENET_STANDARD_STD


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase       ,_lowerCamelCase = PILImageResampling.BICUBIC       ,_lowerCamelCase = None       ,**_lowerCamelCase       ,)							->							np.ndarray:







				'''simple docstring'''





				__lowercase			  =							get_size_dict(_lowerCamelCase       ,default_to_square=_lowerCamelCase	)
				if "shortest_edge" not in size:
							raise ValueError(f"The `size` parameter must contain the key `shortest_edge`. Got {size.keys()}"	)
				__lowercase			  =							get_resize_output_image_size(_lowerCamelCase       ,size=size['''shortest_edge''']       ,default_to_square=_lowerCamelCase	)
				return resize(_lowerCamelCase       ,size=_lowerCamelCase       ,resample=_lowerCamelCase       ,data_format=_lowerCamelCase       ,**_lowerCamelCase	)


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase       ,_lowerCamelCase = None       ,**_lowerCamelCase       ,)							->							np.ndarray:







				'''simple docstring'''





				__lowercase			  =							get_size_dict(_lowerCamelCase	)
				if "height" not in size or "width" not in size:
							raise ValueError(f"The `size` parameter must contain the keys `height` and `width`. Got {size.keys()}"	)
				return center_crop(_lowerCamelCase       ,size=(size['''height'''], size['''width'''])       ,data_format=_lowerCamelCase       ,**_lowerCamelCase	)


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase       ,_lowerCamelCase = None       ,**_lowerCamelCase	)							->							np.ndarray:







				'''simple docstring'''





				return rescale(_lowerCamelCase       ,scale=_lowerCamelCase       ,data_format=_lowerCamelCase       ,**_lowerCamelCase	)


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase       ,_lowerCamelCase       ,_lowerCamelCase = None       ,**_lowerCamelCase       ,)							->							np.ndarray:







				'''simple docstring'''





				return normalize(_lowerCamelCase       ,mean=_lowerCamelCase       ,std=_lowerCamelCase       ,data_format=_lowerCamelCase       ,**_lowerCamelCase	)


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = None       ,_lowerCamelCase = ChannelDimension.FIRST       ,**_lowerCamelCase       ,)							->							Any:







				'''simple docstring'''





				__lowercase			  =							do_resize if do_resize is not None else self.do_resize
				__lowercase			  =							size if size is not None else self.size
				__lowercase			  =							get_size_dict(_lowerCamelCase       ,default_to_square=_lowerCamelCase	)
				__lowercase			  =							resample if resample is not None else self.resample
				__lowercase			  =							do_center_crop if do_center_crop is not None else self.do_center_crop
				__lowercase			  =							crop_size if crop_size is not None else self.crop_size
				__lowercase			  =							get_size_dict(_lowerCamelCase       ,param_name='''crop_size'''	)
				__lowercase			  =							do_rescale if do_rescale is not None else self.do_rescale
				__lowercase			  =							rescale_factor if rescale_factor is not None else self.rescale_factor
				__lowercase			  =							do_normalize if do_normalize is not None else self.do_normalize
				__lowercase			  =							image_mean if image_mean is not None else self.image_mean
				__lowercase			  =							image_std if image_std is not None else self.image_std

				__lowercase			  =							make_list_of_images(_lowerCamelCase	)

				if not valid_images(_lowerCamelCase	):
							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:
							raise ValueError('''Size must be specified if do_resize is True.'''	)

				if do_center_crop and crop_size is None:
							raise ValueError('''Crop size 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.
				__lowercase			  =							[to_numpy_array(_lowerCamelCase	) for image in images]

				if do_resize:
							__lowercase			  =							[self.resize(image=_lowerCamelCase       ,size=_lowerCamelCase       ,resample=_lowerCamelCase	) for image in images]

				if do_center_crop:
							__lowercase			  =							[self.center_crop(image=_lowerCamelCase       ,size=_lowerCamelCase	) for image in images]

				if do_rescale:
							__lowercase			  =							[self.rescale(image=_lowerCamelCase       ,scale=_lowerCamelCase	) for image in images]

				if do_normalize:
							__lowercase			  =							[self.normalize(image=_lowerCamelCase       ,mean=_lowerCamelCase       ,std=_lowerCamelCase	) for image in images]

				__lowercase			  =							[to_channel_dimension_format(_lowerCamelCase       ,_lowerCamelCase	) for image in images]

				__lowercase			  =							{'''pixel_values''': images}
				return BatchFeature(data=_lowerCamelCase       ,tensor_type=_lowerCamelCase	)


	def   _UpperCAmelCase       (self       ,_lowerCamelCase       ,_lowerCamelCase = None	)							->							str:







				'''simple docstring'''





				__lowercase			  =							outputs.logits

				# Resize logits and compute semantic segmentation maps
				if target_sizes is not None:
							if len(_lowerCamelCase	) != len(_lowerCamelCase	):
										raise ValueError(
										    '''Make sure that you pass in as many target sizes as the batch dimension of the logits'''	)

							if is_torch_tensor(_lowerCamelCase	):
										__lowercase			  =							target_sizes.numpy()

							__lowercase			  =							[]

							for idx in range(len(_lowerCamelCase	)	):
										__lowercase			  =							torch.nn.functional.interpolate(
										    logits[idx].unsqueeze(dim=0	)       ,size=target_sizes[idx]       ,mode='''bilinear'''       ,align_corners=_lowerCamelCase	)
										__lowercase			  =							resized_logits[0].argmax(dim=0	)
										semantic_segmentation.append(_lowerCamelCase	)
				else:
							__lowercase			  =							logits.argmax(dim=1	)
							__lowercase			  =							[semantic_segmentation[i] for i in range(semantic_segmentation.shape[0]	)]

				return semantic_segmentation





