





"""simple docstring"""





import math
from typing import Optional

import numpy as np

from ...configuration_utils import PretrainedConfig
from ...utils import logging


_UpperCamelCase					      =   logging.get_logger(__name__)

_UpperCamelCase					      =   {
    'facebook/encodec_24khz': 'https://huggingface.co/facebook/encodec_24khz/resolve/main/config.json',
    'facebook/encodec_48khz': 'https://huggingface.co/facebook/encodec_48khz/resolve/main/config.json',
}
class 		SCREAMING_SNAKE_CASE_				(     snake_case__ ):
				"""simple docstring"""
				__snake_case :							Optional[Any]									=		"""encodec"""




				def __init__(  self	:int    ,     __lowercase	:Dict=[1.5, 3.0, 6.0, 12.0, 24.0]    ,     __lowercase	:int=2_4000    ,     __lowercase	:str=1    ,     __lowercase	:int=False    ,     __lowercase	:List[Any]=None    ,     __lowercase	:Tuple=None    ,     __lowercase	:int=128    ,     __lowercase	:List[Any]=32    ,     __lowercase	:List[str]=1    ,     __lowercase	:Tuple=[8, 5, 4, 2]    ,     __lowercase	:Dict="weight_norm"    ,     __lowercase	:str=7    ,     __lowercase	:Union[str, Any]=7    ,     __lowercase	:Any=3    ,     __lowercase	:Union[str, Any]=2    ,     __lowercase	:str=True    ,     __lowercase	:str="reflect"    ,     __lowercase	:int=2    ,     __lowercase	:Optional[int]=2    ,     __lowercase	:Dict=1.0    ,     __lowercase	:Union[str, Any]=1024    ,     __lowercase	:Dict=None    ,     __lowercase	:Optional[int]=True    ,     **__lowercase	:Dict    ,     ):
											__lowerCamelCase   :   Optional[Any]       =target_bandwidths
											__lowerCamelCase   :   Optional[Any]       =sampling_rate
											__lowerCamelCase   :   Optional[int]       =audio_channels
											__lowerCamelCase   :   int       =normalize
											__lowerCamelCase   :   Any       =chunk_length_s
											__lowerCamelCase   :   str       =overlap
											__lowerCamelCase   :   Optional[int]       =hidden_size
											__lowerCamelCase   :   Optional[int]       =num_filters
											__lowerCamelCase   :   Any       =num_residual_layers
											__lowerCamelCase   :   Any       =upsampling_ratios
											__lowerCamelCase   :   str       =norm_type
											__lowerCamelCase   :   str       =kernel_size
											__lowerCamelCase   :   Optional[Any]       =last_kernel_size
											__lowerCamelCase   :   Tuple       =residual_kernel_size
											__lowerCamelCase   :   Union[str, Any]       =dilation_growth_rate
											__lowerCamelCase   :   str       =use_causal_conv
											__lowerCamelCase   :   str       =pad_mode
											__lowerCamelCase   :   int       =compress
											__lowerCamelCase   :   Any       =num_lstm_layers
											__lowerCamelCase   :   Tuple       =trim_right_ratio
											__lowerCamelCase   :   int       =codebook_size
											__lowerCamelCase   :   Optional[Any]       =codebook_dim if codebook_dim is not None else hidden_size
											__lowerCamelCase   :   Tuple       =use_conv_shortcut

											if self.norm_type not in ["weight_norm", "time_group_norm"]:
																		raise ValueError(
																		    f'self.norm_type must be one of `"weight_norm"`, `"time_group_norm"`), got {self.norm_type}'      )

											super().__init__(**__lowercase      )




				@property
				def        __lowercase	(  self	:List[Any]      ):
											if self.chunk_length_s is None:
																		return None
											else:
																		return int(self.chunk_length_s * self.sampling_rate      )




				@property
				def        __lowercase	(  self	:Optional[Any]      ):
											if self.chunk_length_s is None or self.overlap is None:
																		return None
											else:
																		return max(1    ,     int((1.0 - self.overlap) * self.chunk_length      )      )




				@property
				def        __lowercase	(  self	:Optional[Any]      ):
											__lowerCamelCase   :   Optional[int]       =np.prod(self.upsampling_ratios      )
											return math.ceil(self.sampling_rate / hop_length      )




				@property
				def        __lowercase	(  self	:Union[str, Any]      ):
											return int(1000 * self.target_bandwidths[-1] // (self.frame_rate * 10)      )











"""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


_UpperCamelCase					      =   logging.get_logger(__name__)
class 		SCREAMING_SNAKE_CASE_				(     snake_case__ ):
				"""simple docstring"""
				__snake_case :							Optional[int]									=		["""pixel_values"""]




				def __init__(  self	:Optional[Any]    ,     __lowercase	:bool = True    ,     __lowercase	:Optional[Dict[str, int]] = None    ,     __lowercase	:PILImageResampling = PILImageResampling.BILINEAR    ,     __lowercase	:bool = True    ,     __lowercase	:Dict[str, int] = None    ,     __lowercase	:bool = True    ,     __lowercase	:Union[int, float] = 1 / 255    ,     __lowercase	:bool = True    ,     __lowercase	:Optional[Union[float, List[float]]] = None    ,     __lowercase	:Optional[Union[float, List[float]]] = None    ,     **__lowercase	:Tuple    ,     ):
											super().__init__(**__lowercase      )
											__lowerCamelCase   :   int       =size if size is not None else {'''shortest_edge''': 256}
											__lowerCamelCase   :   int       =get_size_dict(__lowercase    ,     default_to_square=__lowercase      )
											__lowerCamelCase   :   Optional[Any]       =crop_size if crop_size is not None else {'''height''': 224, '''width''': 224}
											__lowerCamelCase   :   List[str]       =get_size_dict(__lowercase    ,     param_name='''crop_size'''      )
											__lowerCamelCase   :   List[str]       =do_resize
											__lowerCamelCase   :   Optional[Any]       =size
											__lowerCamelCase   :   Tuple       =resample
											__lowerCamelCase   :   int       =do_center_crop
											__lowerCamelCase   :   Optional[Any]       =crop_size
											__lowerCamelCase   :   int       =do_rescale
											__lowerCamelCase   :   str       =rescale_factor
											__lowerCamelCase   :   int       =do_normalize
											__lowerCamelCase   :   Union[str, Any]       =image_mean if image_mean is not None else IMAGENET_STANDARD_MEAN
											__lowerCamelCase   :   Tuple       =image_std if image_std is not None else IMAGENET_STANDARD_STD




				def        __lowercase	(  self	:Union[str, Any]    ,     __lowercase	:np.ndarray    ,     __lowercase	:Dict[str, int]    ,     __lowercase	:PILImageResampling = PILImageResampling.BICUBIC    ,     __lowercase	:Optional[Union[str, ChannelDimension]] = None    ,     **__lowercase	:str    ,     ):
											__lowerCamelCase   :   Tuple       =get_size_dict(__lowercase    ,     default_to_square=__lowercase      )
											if "shortest_edge" not in size:
																		raise ValueError(f'The `size` parameter must contain the key `shortest_edge`. Got {size.keys()}'      )
											__lowerCamelCase   :   List[Any]       =get_resize_output_image_size(__lowercase    ,     size=size['''shortest_edge''']    ,     default_to_square=__lowercase      )
											return resize(__lowercase    ,     size=__lowercase    ,     resample=__lowercase    ,     data_format=__lowercase    ,     **__lowercase      )




				def        __lowercase	(  self	:Union[str, Any]    ,     __lowercase	:np.ndarray    ,     __lowercase	:Dict[str, int]    ,     __lowercase	:Optional[Union[str, ChannelDimension]] = None    ,     **__lowercase	:int    ,     ):
											__lowerCamelCase   :   List[Any]       =get_size_dict(__lowercase      )
											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(__lowercase    ,     size=(size['''height'''], size['''width'''])    ,     data_format=__lowercase    ,     **__lowercase      )




				def        __lowercase	(  self	:str    ,     __lowercase	:np.ndarray    ,     __lowercase	:float    ,     __lowercase	:Optional[Union[str, ChannelDimension]] = None    ,     **__lowercase	:str      ):
											return rescale(__lowercase    ,     scale=__lowercase    ,     data_format=__lowercase    ,     **__lowercase      )




				def        __lowercase	(  self	:Union[str, Any]    ,     __lowercase	:np.ndarray    ,     __lowercase	:Union[float, List[float]]    ,     __lowercase	:Union[float, List[float]]    ,     __lowercase	:Optional[Union[str, ChannelDimension]] = None    ,     **__lowercase	:Union[str, Any]    ,     ):
											return normalize(__lowercase    ,     mean=__lowercase    ,     std=__lowercase    ,     data_format=__lowercase    ,     **__lowercase      )




				def        __lowercase	(  self	:List[str]    ,     __lowercase	:ImageInput    ,     __lowercase	:Optional[bool] = None    ,     __lowercase	:Dict[str, int] = None    ,     __lowercase	:PILImageResampling = None    ,     __lowercase	:bool = None    ,     __lowercase	:Dict[str, int] = None    ,     __lowercase	:Optional[bool] = None    ,     __lowercase	:Optional[float] = None    ,     __lowercase	:Optional[bool] = None    ,     __lowercase	:Optional[Union[float, List[float]]] = None    ,     __lowercase	:Optional[Union[float, List[float]]] = None    ,     __lowercase	:Optional[Union[str, TensorType]] = None    ,     __lowercase	:Union[str, ChannelDimension] = ChannelDimension.FIRST    ,     **__lowercase	:Optional[Any]    ,     ):
											__lowerCamelCase   :   Optional[Any]       =do_resize if do_resize is not None else self.do_resize
											__lowerCamelCase   :   Optional[Any]       =size if size is not None else self.size
											__lowerCamelCase   :   List[str]       =get_size_dict(__lowercase    ,     default_to_square=__lowercase      )
											__lowerCamelCase   :   List[str]       =resample if resample is not None else self.resample
											__lowerCamelCase   :   Optional[int]       =do_center_crop if do_center_crop is not None else self.do_center_crop
											__lowerCamelCase   :   Any       =crop_size if crop_size is not None else self.crop_size
											__lowerCamelCase   :   Union[str, Any]       =get_size_dict(__lowercase    ,     param_name='''crop_size'''      )
											__lowerCamelCase   :   Any       =do_rescale if do_rescale is not None else self.do_rescale
											__lowerCamelCase   :   List[str]       =rescale_factor if rescale_factor is not None else self.rescale_factor
											__lowerCamelCase   :   List[Any]       =do_normalize if do_normalize is not None else self.do_normalize
											__lowerCamelCase   :   List[Any]       =image_mean if image_mean is not None else self.image_mean
											__lowerCamelCase   :   int       =image_std if image_std is not None else self.image_std

											__lowerCamelCase   :   int       =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:
																		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.
											__lowerCamelCase   :   int       =[to_numpy_array(__lowercase      ) for image in images]

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

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

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

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

											__lowerCamelCase   :   List[str]       =[to_channel_dimension_format(__lowercase    ,     __lowercase      ) for image in images]

											__lowerCamelCase   :   Union[str, Any]       ={'''pixel_values''': images}
											return BatchFeature(data=__lowercase    ,     tensor_type=__lowercase      )




				def        __lowercase	(  self	:Dict    ,     __lowercase	:Dict    ,     __lowercase	:List[Tuple] = None      ):
											__lowerCamelCase   :   Dict       =outputs.logits

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

																		if is_torch_tensor(__lowercase      ):
																									__lowerCamelCase   :   Optional[int]       =target_sizes.numpy()

																		__lowerCamelCase   :   List[Any]       =[]

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

											return semantic_segmentation




