g4dZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZd d lmZe jeZGd d eZGdde ZdS)zDETR model configuration OrderedDict)Mapping)version)PretrainedConfig) OnnxConfig)logging) verify_backbone_config_arguments)CONFIG_MAPPINGceZdZdZdZdgZdddZ d fd Zede fdZ ede fdZ e de fdZxZS)! DetrConfigax This is the configuration class to store the configuration of a [`DetrModel`]. It is used to instantiate a DETR 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 DETR [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: use_timm_backbone (`bool`, *optional*, defaults to `True`): Whether or not to use the `timm` library for the backbone. If set to `False`, will use the [`AutoBackbone`] API. backbone_config (`PretrainedConfig` or `dict`, *optional*): The configuration of the backbone model. Only used in case `use_timm_backbone` is set to `False` in which case it will default to `ResNetConfig()`. num_channels (`int`, *optional*, defaults to 3): The number of input channels. num_queries (`int`, *optional*, defaults to 100): Number of object queries, i.e. detection slots. This is the maximal number of objects [`DetrModel`] can detect in a single image. For COCO, we recommend 100 queries. d_model (`int`, *optional*, defaults to 256): Dimension of the layers. encoder_layers (`int`, *optional*, defaults to 6): Number of encoder layers. decoder_layers (`int`, *optional*, defaults to 6): Number of decoder layers. encoder_attention_heads (`int`, *optional*, defaults to 8): Number of attention heads for each attention layer in the Transformer encoder. decoder_attention_heads (`int`, *optional*, defaults to 8): Number of attention heads for each attention layer in the Transformer decoder. decoder_ffn_dim (`int`, *optional*, defaults to 2048): Dimension of the "intermediate" (often named feed-forward) layer in decoder. encoder_ffn_dim (`int`, *optional*, defaults to 2048): Dimension of the "intermediate" (often named feed-forward) layer in decoder. activation_function (`str` or `function`, *optional*, defaults to `"relu"`): The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`, `"relu"`, `"silu"` and `"gelu_new"` are supported. dropout (`float`, *optional*, defaults to 0.1): The dropout probability for all fully connected layers in the embeddings, encoder, and pooler. attention_dropout (`float`, *optional*, defaults to 0.0): The dropout ratio for the attention probabilities. activation_dropout (`float`, *optional*, defaults to 0.0): The dropout ratio for activations inside the fully connected layer. init_std (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. init_xavier_std (`float`, *optional*, defaults to 1): The scaling factor used for the Xavier initialization gain in the HM Attention map module. encoder_layerdrop (`float`, *optional*, defaults to 0.0): The LayerDrop probability for the encoder. See the [LayerDrop paper](see https://arxiv.org/abs/1909.11556) for more details. decoder_layerdrop (`float`, *optional*, defaults to 0.0): The LayerDrop probability for the decoder. See the [LayerDrop paper](see https://arxiv.org/abs/1909.11556) for more details. auxiliary_loss (`bool`, *optional*, defaults to `False`): Whether auxiliary decoding losses (loss at each decoder layer) are to be used. position_embedding_type (`str`, *optional*, defaults to `"sine"`): Type of position embeddings to be used on top of the image features. One of `"sine"` or `"learned"`. backbone (`str`, *optional*, defaults to `"resnet50"`): 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*, `True`): 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. dilation (`bool`, *optional*, defaults to `False`): Whether to replace stride with dilation in the last convolutional block (DC5). Only supported when `use_timm_backbone` = `True`. class_cost (`float`, *optional*, defaults to 1): Relative weight of the classification error in the Hungarian matching cost. bbox_cost (`float`, *optional*, defaults to 5): Relative weight of the L1 error of the bounding box coordinates in the Hungarian matching cost. giou_cost (`float`, *optional*, defaults to 2): Relative weight of the generalized IoU loss of the bounding box in the Hungarian matching cost. mask_loss_coefficient (`float`, *optional*, defaults to 1): Relative weight of the Focal loss in the panoptic segmentation loss. dice_loss_coefficient (`float`, *optional*, defaults to 1): Relative weight of the DICE/F-1 loss in the panoptic segmentation loss. bbox_loss_coefficient (`float`, *optional*, defaults to 5): Relative weight of the L1 bounding box loss in the object detection loss. giou_loss_coefficient (`float`, *optional*, defaults to 2): Relative weight of the generalized IoU loss in the object detection loss. eos_coefficient (`float`, *optional*, defaults to 0.1): Relative classification weight of the 'no-object' class in the object detection loss. Examples: ```python >>> from transformers import DetrConfig, DetrModel >>> # Initializing a DETR facebook/detr-resnet-50 style configuration >>> configuration = DetrConfig() >>> # Initializing a model (with random weights) from the facebook/detr-resnet-50 style configuration >>> model = DetrModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```detrpast_key_valuesd_modelencoder_attention_heads) hidden_sizenum_attention_headsTNrdrelu皙?{Gz??Fsineresnet50r c# |r|i}|rd|d<gd|d<||d<n|s|dvr|2tdtdd g }nLt|tr7|d }$t|$}%|%|}d}d}t||||| ||_||_ ||_ ||_ ||_ ||_ ||_||_| |_||_| |_||_||_||_||_||_||_| |_| |_||_||_||_||_||_||_ ||_!||_"||_#||_$||_%||_&| |_'|!|_(|"|_)tUj+dd | i|#dS)N output_stride)r"r r out_indicesin_chans)Nr!zX`backbone_config` is `None`. Initializing the config with the default `ResNet` backbone.resnetstage4) out_features model_type)use_timm_backboneuse_pretrained_backbonebackbonebackbone_configbackbone_kwargsis_encoder_decoder),loggerinfor isinstancedictget from_dictr r.r1 num_channels num_queriesrencoder_ffn_dimencoder_layersrdecoder_ffn_dimdecoder_layersdecoder_attention_headsdropoutattention_dropoutactivation_dropoutactivation_functioninit_stdinit_xavier_stdencoder_layerdropdecoder_layerdropnum_hidden_layersauxiliary_lossposition_embedding_typer0r/r2dilation class_cost bbox_cost giou_costmask_loss_coefficientdice_loss_coefficientbbox_loss_coefficientgiou_loss_coefficienteos_coefficientsuper__init__)'selfr.r1r;r<r>r=rr@r?rArHrIr3rErrBrCrDrFrGrKrLr0r/r2rMrNrOrPrQrRrSrTrUkwargsbackbone_model_type config_class __class__s' g/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/detr/configuration_detr.pyrWzDetrConfig.__init__s2P  !8 O 6350-9\\OM **6OJ ' '" x3E'E'E& vwww"0": "S"S"SOT22 J&5&9&9,&G&G#-.AB "."8"8"I"IHH(/$;++     "3.(& .,'>$.,'>$ !2"4#6   .!2!2!/,'>$  '>$.  $""%:"%:"%:"%:".II,>I&IIIIIreturnc|jSN)rrXs r]rzDetrConfig.num_attention_headss ++r^c|jSra)rrbs r]rzDetrConfig.hidden_sizes |r^r1c |dd|i|S)a-Instantiate a [`DetrConfig`] (or a derived class) from a pre-trained backbone model configuration. Args: backbone_config ([`PretrainedConfig`]): The backbone configuration. Returns: [`DetrConfig`]: An instance of a configuration object r1r4r4)clsr1rYs r]from_backbone_configzDetrConfig.from_backbone_configss==?=f===r^)"TNrrrrrrrrrrTrrrrrrrFr r!TNFr"r#r r"r"r#r r)__name__ __module__ __qualname____doc__r-keys_to_ignore_at_inference attribute_maprWpropertyintrr classmethodrrf __classcell__)r\s@r]rr sXd d LJ#4"5 8M ! !" & $GgJgJgJgJgJgJR,S,,,X,SX >3C > > >[ > > > > >r^rceZdZejdZedeeee efffdZ ede fdZ ede fdZ dS)DetrOnnxConfigz1.11r_c:tddddddfdddifgS) N pixel_valuesbatchr;heightwidth)rr"r r pixel_maskrrrbs r]inputszDetrOnnxConfig.inputss:WHQX!Y!YZ7|,    r^cdS)Ngh㈵>r4rbs r]atol_for_validationz"DetrOnnxConfig.atol_for_validationstr^cdS)N r4rbs r]default_onnx_opsetz!DetrOnnxConfig.default_onnx_opsetsrr^N)rgrhrirparsetorch_onnx_minimum_versionrmrstrrnryfloatr{r~r4r^r]rrrr s!.v!6!6  WS#X%6 67   X UXCXr^rrN)rj collectionsrtypingr packagingrconfiguration_utilsronnxr utilsr utils.backbone_utilsr autor get_loggerrgr5rrrr4r^r]rs######333333DDDDDD!!!!!!  H % %i>i>i>i>i>!i>i>i>XZr^