g1rdZddlmZddlmZddlmZejeZ ddiZ Gdd eZ d S) zZoeDepth model configuration)PretrainedConfig)logging)CONFIG_MAPPINGzIntel/zoedepth-nyuzBhttps://huggingface.co/Intel/zoedepth-nyu/resolve/main/config.jsonczeZdZdZdZddddddddgd gd d d ddd dd gddddddddddddgddddffd ZxZS)ZoeDepthConfiga This is the configuration class to store the configuration of a [`ZoeDepthForDepthEstimation`]. It is used to instantiate an ZoeDepth model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the ZoeDepth [Intel/zoedepth-nyu](https://huggingface.co/Intel/zoedepth-nyu) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: backbone_config (`Union[Dict[str, Any], PretrainedConfig]`, *optional*, defaults to `BeitConfig()`): The configuration of the backbone model. backbone (`str`, *optional*): Name of backbone to use when `backbone_config` is `None`. If `use_pretrained_backbone` is `True`, this will load the corresponding pretrained weights from the timm or transformers library. If `use_pretrained_backbone` is `False`, this loads the backbone's config and uses that to initialize the backbone with random weights. use_pretrained_backbone (`bool`, *optional*, defaults to `False`): Whether to use pretrained weights for the backbone. backbone_kwargs (`dict`, *optional*): Keyword arguments to be passed to AutoBackbone when loading from a checkpoint e.g. `{'out_indices': (0, 1, 2, 3)}`. Cannot be specified if `backbone_config` is set. hidden_act (`str` or `function`, *optional*, defaults to `"gelu"`): The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`, `"relu"`, `"selu"` and `"gelu_new"` are supported. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. batch_norm_eps (`float`, *optional*, defaults to 1e-05): The epsilon used by the batch normalization layers. readout_type (`str`, *optional*, defaults to `"project"`): The readout type to use when processing the readout token (CLS token) of the intermediate hidden states of the ViT backbone. Can be one of [`"ignore"`, `"add"`, `"project"`]. - "ignore" simply ignores the CLS token. - "add" passes the information from the CLS token to all other tokens by adding the representations. - "project" passes information to the other tokens by concatenating the readout to all other tokens before projecting the representation to the original feature dimension D using a linear layer followed by a GELU non-linearity. reassemble_factors (`List[int]`, *optional*, defaults to `[4, 2, 1, 0.5]`): The up/downsampling factors of the reassemble layers. neck_hidden_sizes (`List[str]`, *optional*, defaults to `[96, 192, 384, 768]`): The hidden sizes to project to for the feature maps of the backbone. fusion_hidden_size (`int`, *optional*, defaults to 256): The number of channels before fusion. head_in_index (`int`, *optional*, defaults to -1): The index of the features to use in the heads. use_batch_norm_in_fusion_residual (`bool`, *optional*, defaults to `False`): Whether to use batch normalization in the pre-activate residual units of the fusion blocks. use_bias_in_fusion_residual (`bool`, *optional*, defaults to `True`): Whether to use bias in the pre-activate residual units of the fusion blocks. num_relative_features (`int`, *optional*, defaults to 32): The number of features to use in the relative depth estimation head. add_projection (`bool`, *optional*, defaults to `False`): Whether to add a projection layer before the depth estimation head. bottleneck_features (`int`, *optional*, defaults to 256): The number of features in the bottleneck layer. num_attractors (`List[int], *optional*, defaults to `[16, 8, 4, 1]`): The number of attractors to use in each stage. bin_embedding_dim (`int`, *optional*, defaults to 128): The dimension of the bin embeddings. attractor_alpha (`int`, *optional*, defaults to 1000): The alpha value to use in the attractor. attractor_gamma (`int`, *optional*, defaults to 2): The gamma value to use in the attractor. attractor_kind (`str`, *optional*, defaults to `"mean"`): The kind of attractor to use. Can be one of [`"mean"`, `"sum"`]. min_temp (`float`, *optional*, defaults to 0.0212): The minimum temperature value to consider. max_temp (`float`, *optional*, defaults to 50.0): The maximum temperature value to consider. bin_centers_type (`str`, *optional*, defaults to `"softplus"`): Activation type used for bin centers. Can be "normed" or "softplus". For "normed" bin centers, linear normalization trick is applied. This results in bounded bin centers. For "softplus", softplus activation is used and thus are unbounded. bin_configurations (`List[dict]`, *optional*, defaults to `[{'n_bins': 64, 'min_depth': 0.001, 'max_depth': 10.0}]`): Configuration for each of the bin heads. Each configuration should consist of the following keys: - name (`str`): The name of the bin head - only required in case of multiple bin configurations. - `n_bins` (`int`): The number of bins to use. - `min_depth` (`float`): The minimum depth value to consider. - `max_depth` (`float`): The maximum depth value to consider. In case only a single configuration is passed, the model will use a single head with the specified configuration. In case multiple configurations are passed, the model will use multiple heads with the specified configurations. num_patch_transformer_layers (`int`, *optional*): The number of transformer layers to use in the patch transformer. Only used in case of multiple bin configurations. patch_transformer_hidden_size (`int`, *optional*): The hidden size to use in the patch transformer. Only used in case of multiple bin configurations. patch_transformer_intermediate_size (`int`, *optional*): The intermediate size to use in the patch transformer. Only used in case of multiple bin configurations. patch_transformer_num_attention_heads (`int`, *optional*): The number of attention heads to use in the patch transformer. Only used in case of multiple bin configurations. Example: ```python >>> from transformers import ZoeDepthConfig, ZoeDepthForDepthEstimation >>> # Initializing a ZoeDepth zoedepth-large style configuration >>> configuration = ZoeDepthConfig() >>> # Initializing a model from the zoedepth-large style configuration >>> model = ZoeDepthForDepthEstimation(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```zoedepthNFgelug{Gz?gh㈵>project)rg?)`i )r r irmeanggs?gI@softplus@gMbP?g$@)n_bins min_depth max_depthc tjdi||dvrtd|dvrtd|rtd||td|<|:tdt dd d d d d ddgd}nLt |tr7|d} t | }!|! |}||r|td||_ ||_ ||_ ||_ ||_||_||_| |_| |_| |_| |_| |_||_||_||_||_||_||_||_||_||_||_||_ ||_!||_"||_#||_$||_%||_&dS)N)ignoreaddr z8Readout_type must be one of ['ignore', 'add', 'project'])rsumz-Attractor_kind must be one of ['mean', 'sum']z+Pretrained backbones are not supported yet.z8You can't specify both `backbone` and `backbone_config`.zV`backbone_config` is `None`. Initializing the config with the default `BEiT` backbone.beitriirTF)stage6stage12stage18stage24) image_sizenum_hidden_layers hidden_sizeintermediate_sizenum_attention_headsuse_relative_position_biasreshape_hidden_states out_features model_typez?You can't specify both `backbone_kwargs` and `backbone_config`.)'super__init__ ValueErrorloggerinfor isinstancedictget from_dictbackbone_configbackbone hidden_actuse_pretrained_backboneinitializer_rangebatch_norm_eps readout_typereassemble_factorsneck_hidden_sizesfusion_hidden_size head_in_index!use_batch_norm_in_fusion_residualuse_bias_in_fusion_residualnum_relative_featuresadd_projectionbottleneck_featuresnum_attractorsbin_embedding_dimattractor_alphaattractor_gammaattractor_kindmin_tempmax_tempbin_centers_typebin_configurationsnum_patch_transformer_layerspatch_transformer_hidden_size#patch_transformer_intermediate_size%patch_transformer_num_attention_heads)#selfr:r;r=backbone_kwargsr<r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVkwargsbackbone_model_type config_class __class__s# o/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/zoedepth/configuration_zoedepth.pyr2zZoeDepthConfig.__init__s1D ""6""" ; ; ;WXX X  0 0LMM M " LJKK K  &8+?WXX X  "x'7 KKp q q q,V4"$ "&$&+/&+HHH   OO . . F"1"5"5l"C"C )*=>L*44_EEO  &? &?Z^__ _.  $'>$!2,("4!2"4*1R.+F(%:",#6 ,!2..,    0"4,H)-J*3V05Z222)__name__ __module__ __qualname____doc__r/r2 __classcell__)r\s@r]rrsg g RJ %)>>---*/$( $}}#')DQQR%)&*,0.2?a[a[a[a[a[a[a[a[a[a[r^rN) rbconfiguration_utilsrutilsrauto.configuration_autor get_loggerr_r4&ZOEDEPTH_PRETRAINED_CONFIG_ARCHIVE_MAPrr0r^r]ris#"333333444444  H % %^*& M[M[M[M[M[%M[M[M[M[M[r^