
"""simple docstring"""


import unittest

import numpy as np

from transformers.file_utils import is_torch_available, is_vision_available
from transformers.testing_utils import require_torch, require_vision

from ...test_image_processing_common import ImageProcessingSavingTestMixin, prepare_image_inputs


if is_torch_available():
		import torch

if is_vision_available():
		from PIL import Image

		from transformers import DPTImageProcessor







class 					lowerCAmelCase_					(   unittest.TestCase     ):



						'''simple docstring'''


						def __init__(			self  :   List[Any]	,A_  :   Tuple	,A_  :   Dict=7	,A_  :   List[Any]=3	,A_  :   int=18	,A_  :   Any=30	,A_  :   Tuple=400	,A_  :   str=True	,A_  :   List[Any]=None	,A_  :   str=True	,A_  :   Any=[0.5, 0.5, 0.5]	,A_  :   Union[str, Any]=[0.5, 0.5, 0.5]	,)  -> Any:
							A     = size if size is not None else {"""height""": 18, """width""": 18}
							A     = parent
							A     = batch_size
							A     = num_channels
							A     = image_size
							A     = min_resolution
							A     = max_resolution
							A     = do_resize
							A     = size
							A     = do_normalize
							A     = image_mean
							A     = image_std




						def     _SCREAMING_SNAKE_CASE   (			self  :   List[str]   )  -> List[str]:
							return {
							    "image_mean": self.image_mean,
							    "image_std": self.image_std,
							    "do_normalize": self.do_normalize,
							    "do_resize": self.do_resize,
							    "size": self.size,
							}







@require_torch
@require_vision
class 					lowerCAmelCase_					(   a_		,       unittest.TestCase     ):



						'''simple docstring'''


						_lowerCamelCase:     List[Any]   =     DPTImageProcessor if is_vision_available() else None


						def     _SCREAMING_SNAKE_CASE   (			self  :   str   )  -> Any:
							A     = DPTImageProcessingTester(self   )


						@property
						def     _SCREAMING_SNAKE_CASE   (			self  :   Tuple   )  -> str:
							return self.image_processor_tester.prepare_image_processor_dict()


						def     _SCREAMING_SNAKE_CASE   (			self  :   Union[str, Any]   )  -> List[Any]:
							A     = self.image_processing_class(**self.image_processor_dict   )
							self.assertTrue(hasattr(_lowercase	,'image_mean'   )   )
							self.assertTrue(hasattr(_lowercase	,'image_std'   )   )
							self.assertTrue(hasattr(_lowercase	,'do_normalize'   )   )
							self.assertTrue(hasattr(_lowercase	,'do_resize'   )   )
							self.assertTrue(hasattr(_lowercase	,'size'   )   )


						def     _SCREAMING_SNAKE_CASE   (			self  :   Optional[int]   )  -> Optional[int]:
							A     = self.image_processing_class.from_dict(self.image_processor_dict   )
							self.assertEqual(image_processor.size	,{'height': 18, 'width': 18}   )

							A     = self.image_processing_class.from_dict(self.image_processor_dict	,size=42   )
							self.assertEqual(image_processor.size	,{'height': 42, 'width': 42}   )


						def     _SCREAMING_SNAKE_CASE   (			self  :   List[str]   )  -> Optional[int]:
							# Initialize image_processing
							A     = self.image_processing_class(**self.image_processor_dict   )
							# create random PIL images
							A     = prepare_image_inputs(self.image_processor_tester	,equal_resolution=_lowercase   )
							for image in image_inputs:
								self.assertIsInstance(_lowercase	,Image.Image   )

							# Test not batched input
							A     = image_processing(image_inputs[0]	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)

							# Test batched
							A     = image_processing(_lowercase	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)


						def     _SCREAMING_SNAKE_CASE   (			self  :   Union[str, Any]   )  -> List[Any]:
							# Initialize image_processing
							A     = self.image_processing_class(**self.image_processor_dict   )
							# create random numpy tensors
							A     = prepare_image_inputs(self.image_processor_tester	,equal_resolution=_lowercase	,numpify=_lowercase   )
							for image in image_inputs:
								self.assertIsInstance(_lowercase	,np.ndarray   )

							# Test not batched input
							A     = image_processing(image_inputs[0]	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)

							# Test batched
							A     = image_processing(_lowercase	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)




						def     _SCREAMING_SNAKE_CASE   (			self  :   int   )  -> Dict:
							# Initialize image_processing
							A     = self.image_processing_class(**self.image_processor_dict   )
							# create random PyTorch tensors
							A     = prepare_image_inputs(self.image_processor_tester	,equal_resolution=_lowercase	,torchify=_lowercase   )
							for image in image_inputs:
								self.assertIsInstance(_lowercase	,torch.Tensor   )

							# Test not batched input
							A     = image_processing(image_inputs[0]	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        1,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)

							# Test batched
							A     = image_processing(_lowercase	,return_tensors='pt'   ).pixel_values
							self.assertEqual(
							    encoded_images.shape	,(
							        self.image_processor_tester.batch_size,
							        self.image_processor_tester.num_channels,
							        self.image_processor_tester.size['height'],
							        self.image_processor_tester.size['width'],
							    )	,)


import importlib
import json
import os
from collections import OrderedDict
from typing import Dict, Optional, Union

# Build the list of all feature extractors
from ...configuration_utils import PretrainedConfig
from ...dynamic_module_utils import get_class_from_dynamic_module, resolve_trust_remote_code
from ...feature_extraction_utils import FeatureExtractionMixin
from ...utils import CONFIG_NAME, FEATURE_EXTRACTOR_NAME, get_file_from_repo, logging
from .auto_factory import _LazyAutoMapping
from .configuration_auto import (
    CONFIG_MAPPING_NAMES,
    AutoConfig,
    model_type_to_module_name,
    replace_list_option_in_docstrings,
)


__snake_case				:      Union[str, Any]   		=			logging.get_logger(__name__)

__snake_case				:      Tuple   		=			OrderedDict(
    [
        ("""audio-spectrogram-transformer""", """ASTFeatureExtractor"""),
        ("""beit""", """BeitFeatureExtractor"""),
        ("""chinese_clip""", """ChineseCLIPFeatureExtractor"""),
        ("""clap""", """ClapFeatureExtractor"""),
        ("""clip""", """CLIPFeatureExtractor"""),
        ("""clipseg""", """ViTFeatureExtractor"""),
        ("""conditional_detr""", """ConditionalDetrFeatureExtractor"""),
        ("""convnext""", """ConvNextFeatureExtractor"""),
        ("""cvt""", """ConvNextFeatureExtractor"""),
        ("""data2vec-audio""", """Wav2Vec2FeatureExtractor"""),
        ("""data2vec-vision""", """BeitFeatureExtractor"""),
        ("""deformable_detr""", """DeformableDetrFeatureExtractor"""),
        ("""deit""", """DeiTFeatureExtractor"""),
        ("""detr""", """DetrFeatureExtractor"""),
        ("""dinat""", """ViTFeatureExtractor"""),
        ("""donut-swin""", """DonutFeatureExtractor"""),
        ("""dpt""", """DPTFeatureExtractor"""),
        ("""encodec""", """EncodecFeatureExtractor"""),
        ("""flava""", """FlavaFeatureExtractor"""),
        ("""glpn""", """GLPNFeatureExtractor"""),
        ("""groupvit""", """CLIPFeatureExtractor"""),
        ("""hubert""", """Wav2Vec2FeatureExtractor"""),
        ("""imagegpt""", """ImageGPTFeatureExtractor"""),
        ("""layoutlmv2""", """LayoutLMv2FeatureExtractor"""),
        ("""layoutlmv3""", """LayoutLMv3FeatureExtractor"""),
        ("""levit""", """LevitFeatureExtractor"""),
        ("""maskformer""", """MaskFormerFeatureExtractor"""),
        ("""mctct""", """MCTCTFeatureExtractor"""),
        ("""mobilenet_v1""", """MobileNetV1FeatureExtractor"""),
        ("""mobilenet_v2""", """MobileNetV2FeatureExtractor"""),
        ("""mobilevit""", """MobileViTFeatureExtractor"""),
        ("""nat""", """ViTFeatureExtractor"""),
        ("""owlvit""", """OwlViTFeatureExtractor"""),
        ("""perceiver""", """PerceiverFeatureExtractor"""),
        ("""poolformer""", """PoolFormerFeatureExtractor"""),
        ("""regnet""", """ConvNextFeatureExtractor"""),
        ("""resnet""", """ConvNextFeatureExtractor"""),
        ("""segformer""", """SegformerFeatureExtractor"""),
        ("""sew""", """Wav2Vec2FeatureExtractor"""),
        ("""sew-d""", """Wav2Vec2FeatureExtractor"""),
        ("""speech_to_text""", """Speech2TextFeatureExtractor"""),
        ("""speecht5""", """SpeechT5FeatureExtractor"""),
        ("""swiftformer""", """ViTFeatureExtractor"""),
        ("""swin""", """ViTFeatureExtractor"""),
        ("""swinv2""", """ViTFeatureExtractor"""),
        ("""table-transformer""", """DetrFeatureExtractor"""),
        ("""timesformer""", """VideoMAEFeatureExtractor"""),
        ("""tvlt""", """TvltFeatureExtractor"""),
        ("""unispeech""", """Wav2Vec2FeatureExtractor"""),
        ("""unispeech-sat""", """Wav2Vec2FeatureExtractor"""),
        ("""van""", """ConvNextFeatureExtractor"""),
        ("""videomae""", """VideoMAEFeatureExtractor"""),
        ("""vilt""", """ViltFeatureExtractor"""),
        ("""vit""", """ViTFeatureExtractor"""),
        ("""vit_mae""", """ViTFeatureExtractor"""),
        ("""vit_msn""", """ViTFeatureExtractor"""),
        ("""wav2vec2""", """Wav2Vec2FeatureExtractor"""),
        ("""wav2vec2-conformer""", """Wav2Vec2FeatureExtractor"""),
        ("""wavlm""", """Wav2Vec2FeatureExtractor"""),
        ("""whisper""", """WhisperFeatureExtractor"""),
        ("""xclip""", """CLIPFeatureExtractor"""),
        ("""yolos""", """YolosFeatureExtractor"""),
    ]
)

__snake_case				:      List[Any]   		=			_LazyAutoMapping(CONFIG_MAPPING_NAMES, FEATURE_EXTRACTOR_MAPPING_NAMES)


def 					_UpperCAmelCase				(						a__):

			'''simple docstring'''



			for module_name, extractors in FEATURE_EXTRACTOR_MAPPING_NAMES.items():
						if class_name in extractors:
									a_						:						Tuple 							= model_type_to_module_name(a__)

									a_						:						Any 							= importlib.import_module(f'''.{module_name}'''  ,	"""transformers.models""")
									try:
												return getattr(a__  ,	a__)
									except AttributeError:
												continue

			for _, extractor in FEATURE_EXTRACTOR_MAPPING._extra_content.items():
						if getattr(a__  ,	"""__name__"""  ,	a__) == class_name:
									return extractor

    # We did not fine the class, but maybe it's because a dep is missing. In that case, the class will be in the main
    # init and we return the proper dummy to get an appropriate error message.
			a_						:						Tuple 							= importlib.import_module("""transformers""")
			if hasattr(a__  ,	a__):
						return getattr(a__  ,	a__)

			return None


def 					_UpperCAmelCase				(						a__  ,	a__ = None  ,	a__ = False  ,	a__ = False  ,	a__ = None  ,	a__ = None  ,	a__ = None  ,	a__ = False  ,	**a__  ,	):

			'''simple docstring'''



			a_						:						List[str] 							= get_file_from_repo(
			    a__  ,	a__  ,	cache_dir=a__  ,	force_download=a__  ,	resume_download=a__  ,	proxies=a__  ,	use_auth_token=a__  ,	revision=a__  ,	local_files_only=a__  ,	)
			if resolved_config_file is None:
						logger.info(
						    """Could not locate the feature extractor configuration file, will try to use the model config instead.""")
						return {}

			with open(a__  ,	encoding="""utf-8""") as reader:
						return json.load(a__)



class        A__:
						"""simple docstring"""






						def __init__(    self    )     -> List[str]:
									raise EnvironmentError(
									    """AutoFeatureExtractor is designed to be instantiated """
									    """using the `AutoFeatureExtractor.from_pretrained(pretrained_model_name_or_path)` method."""    )





						@classmethod
						@replace_list_option_in_docstrings(_lowercase    )
						def 			UpperCamelCase__					(    cls  ,      _lowercase  ,      **_lowercase    )     -> Dict:
									a_						:						Tuple 							= kwargs.pop("""config"""  ,      _lowercase    )
									a_						:						Dict 							= kwargs.pop("""trust_remote_code"""  ,      _lowercase    )
									a_						:						Dict 							= True

									a_							,					a_						:						List[str] 							= FeatureExtractionMixin.get_feature_extractor_dict(_lowercase  ,      **_lowercase    )
									a_						:						Tuple 							= config_dict.get("""feature_extractor_type"""  ,      _lowercase    )
									a_						:						List[Any] 							= None
									if "AutoFeatureExtractor" in config_dict.get("""auto_map"""  ,      {}    ):
												a_						:						Any 							= config_dict["""auto_map"""]["""AutoFeatureExtractor"""]

									# If we don't find the feature extractor class in the feature extractor config, let's try the model config.
									if feature_extractor_class is None and feature_extractor_auto_map is None:
												if not isinstance(_lowercase  ,      _lowercase    ):
															a_						:						List[str] 							= AutoConfig.from_pretrained(_lowercase  ,      **_lowercase    )
												# It could be in `config.feature_extractor_type``
												a_						:						List[Any] 							= getattr(_lowercase  ,      """feature_extractor_type"""  ,      _lowercase    )
												if hasattr(_lowercase  ,      """auto_map"""    ) and "AutoFeatureExtractor" in config.auto_map:
															a_						:						List[str] 							= config.auto_map["""AutoFeatureExtractor"""]

									if feature_extractor_class is not None:
												a_						:						int 							= feature_extractor_class_from_name(_lowercase    )

									a_						:						Dict 							= feature_extractor_auto_map is not None
									a_						:						Union[str, Any] 							= feature_extractor_class is not None or type(_lowercase    ) in FEATURE_EXTRACTOR_MAPPING
									a_						:						Optional[Any] 							= resolve_trust_remote_code(
									    _lowercase  ,      _lowercase  ,      _lowercase  ,      _lowercase    )

									if has_remote_code and trust_remote_code:
												a_						:						int 							= get_class_from_dynamic_module(
												    _lowercase  ,      _lowercase  ,      **_lowercase    )
												a_						:						Tuple 							= kwargs.pop("""code_revision"""  ,      _lowercase    )
												if os.path.isdir(_lowercase    ):
															feature_extractor_class.register_for_auto_class()
												return feature_extractor_class.from_dict(_lowercase  ,      **_lowercase    )
									elif feature_extractor_class is not None:
												return feature_extractor_class.from_dict(_lowercase  ,      **_lowercase    )
									# Last try: we use the FEATURE_EXTRACTOR_MAPPING.
									elif type(_lowercase    ) in FEATURE_EXTRACTOR_MAPPING:
												a_						:						List[str] 							= FEATURE_EXTRACTOR_MAPPING[type(_lowercase    )]
												return feature_extractor_class.from_dict(_lowercase  ,      **_lowercase    )

									raise ValueError(
									    F'''Unrecognized feature extractor in {pretrained_model_name_or_path}. Should have a '''
									    F'''`feature_extractor_type` key in its {FEATURE_EXTRACTOR_NAME} of {CONFIG_NAME}, or one of the following '''
									    F'''`model_type` keys in its {CONFIG_NAME}: {", ".join(c for c in FEATURE_EXTRACTOR_MAPPING_NAMES.keys()    )}'''    )





						@staticmethod
						def 			UpperCamelCase__					(    _lowercase  ,      _lowercase    )     -> List[Any]:
									FEATURE_EXTRACTOR_MAPPING.register(_lowercase  ,      _lowercase    )


