from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


UpperCamelCase													=		{
    '''configuration_mobilenet_v1''': [
        '''MOBILENET_V1_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''MobileNetV1Config''',
        '''MobileNetV1OnnxConfig''',
    ],
}

try:
      if not is_vision_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase													=		['''MobileNetV1FeatureExtractor''']
      UpperCamelCase													=		['''MobileNetV1ImageProcessor''']

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase													=		[
          '''MOBILENET_V1_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''MobileNetV1ForImageClassification''',
          '''MobileNetV1Model''',
          '''MobileNetV1PreTrainedModel''',
          '''load_tf_weights_in_mobilenet_v1''',
      ]


if TYPE_CHECKING:
      from .configuration_mobilenet_va import (
          MOBILENET_V1_PRETRAINED_CONFIG_ARCHIVE_MAP,
          MobileNetVaConfig,
          MobileNetVaOnnxConfig,
      )

      try:
            if not is_vision_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .feature_extraction_mobilenet_va import MobileNetVaFeatureExtractor
            from .image_processing_mobilenet_va import MobileNetVaImageProcessor

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_mobilenet_va import (
                MOBILENET_V1_PRETRAINED_MODEL_ARCHIVE_LIST,
                MobileNetVaForImageClassification,
                MobileNetVaModel,
                MobileNetVaPreTrainedModel,
                load_tf_weights_in_mobilenet_va,
            )


else:
      import sys

      UpperCamelCase													=		_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)




from collections import OrderedDict
from typing import Mapping

from packaging import version

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


UpperCamelCase													=		logging.get_logger(__name__)

UpperCamelCase													=		{
    '''apple/mobilevit-small''': '''https://huggingface.co/apple/mobilevit-small/resolve/main/config.json''',
    '''apple/mobilevit-x-small''': '''https://huggingface.co/apple/mobilevit-x-small/resolve/main/config.json''',
    '''apple/mobilevit-xx-small''': '''https://huggingface.co/apple/mobilevit-xx-small/resolve/main/config.json''',
    '''apple/deeplabv3-mobilevit-small''': (
        '''https://huggingface.co/apple/deeplabv3-mobilevit-small/resolve/main/config.json'''
    ),
    '''apple/deeplabv3-mobilevit-x-small''': (
        '''https://huggingface.co/apple/deeplabv3-mobilevit-x-small/resolve/main/config.json'''
    ),
    '''apple/deeplabv3-mobilevit-xx-small''': (
        '''https://huggingface.co/apple/deeplabv3-mobilevit-xx-small/resolve/main/config.json'''
    ),
    # See all MobileViT models at https://huggingface.co/models?filter=mobilevit
}



class 						snake_case_ (	__A ):
   __A			:	Any        =       "mobilevit"






   def __init__(   self       :  Union[str, Any]			,		lowercase_       :  List[Any]=3			,		lowercase_       :  List[Any]=2_56			,		lowercase_       :  Union[str, Any]=2			,		lowercase_       :  int=[1_44, 1_92, 2_40]			,		lowercase_       :  List[str]=[16, 32, 64, 96, 1_28, 1_60, 6_40]			,		lowercase_       :  Tuple=4			,		lowercase_       :  Optional[int]=2.0			,		lowercase_       :  int=4.0			,		lowercase_       :  str="silu"			,		lowercase_       :  Optional[Any]=3			,		lowercase_       :  Dict=32			,		lowercase_       :  Union[str, Any]=0.1			,		lowercase_       :  str=0.0			,		lowercase_       :  str=0.1			,		lowercase_       :  List[str]=0.02			,		lowercase_       :  Union[str, Any]=1E-5			,		lowercase_       :  Tuple=True			,		lowercase_       :  Optional[Any]=2_56			,		lowercase_       :  List[str]=[6, 12, 18]			,		lowercase_       :  Any=0.1			,		lowercase_       :  List[str]=2_55			,		**lowercase_       :  Tuple			,		)      ->		Optional[Any]:
       super().__init__(**lowercase_						)

       lowercase__				:     List[str]					      =       num_channels
       lowercase__				:     Optional[int]					      =       image_size
       lowercase__				:     Any					      =       patch_size
       lowercase__				:     int					      =       hidden_sizes
       lowercase__				:     str					      =       neck_hidden_sizes
       lowercase__				:     Optional[int]					      =       num_attention_heads
       lowercase__				:     Tuple					      =       mlp_ratio
       lowercase__				:     Tuple					      =       expand_ratio
       lowercase__				:     List[str]					      =       hidden_act
       lowercase__				:     Optional[int]					      =       conv_kernel_size
       lowercase__				:     Union[str, Any]					      =       output_stride
       lowercase__				:     Optional[int]					      =       hidden_dropout_prob
       lowercase__				:     List[str]					      =       attention_probs_dropout_prob
       lowercase__				:     Union[str, Any]					      =       classifier_dropout_prob
       lowercase__				:     Union[str, Any]					      =       initializer_range
       lowercase__				:     List[str]					      =       layer_norm_eps
       lowercase__				:     Dict					      =       qkv_bias

       # decode head attributes for semantic segmentation
       lowercase__				:     List[Any]					      =       aspp_out_channels
       lowercase__				:     List[Any]					      =       atrous_rates
       lowercase__				:     str					      =       aspp_dropout_prob
       lowercase__				:     int					      =       semantic_loss_ignore_index



class 						snake_case_ (	__A ):
   __A			:	int        =       version.parse("1.11" )






   @property
   def    __UpperCamelCase  (   self       :  Union[str, Any]						)      ->		Mapping[str, Mapping[int, str]]:
       return OrderedDict([("pixel_values", {0: "batch", 1: "num_channels", 2: "height", 3: "width"})]						)






   @property
   def    __UpperCamelCase  (   self       :  str						)      ->		Mapping[str, Mapping[int, str]]:
       if self.task == "image-classification":
           return OrderedDict([("logits", {0: "batch"})]						)
       else:
           return OrderedDict([("last_hidden_state", {0: "batch"}), ("pooler_output", {0: "batch"})]						)







   @property
   def    __UpperCamelCase  (   self       :  Union[str, Any]						)      ->		float:
       return 1E-4



