FPh~ddlmZmZmZddlZddlmZddlmZddl m Z GddejZ Gdd ejZ y) )ListTupleTypeN)nn) functional) LayerNorm2dceZdZdZdej ddddedejdedeejd ed ed d ffd Z de jde jde jde jde d e e je jff dZde jde jde jde jd e e je jff dZxZS) MaskDecoderan Decoder module for generating masks and their associated quality scores, using a transformer architecture to predict masks given image and prompt embeddings. Attributes: transformer_dim (int): Channel dimension for the transformer module. transformer (nn.Module): The transformer module used for mask prediction. num_multimask_outputs (int): Number of masks to predict for disambiguating masks. iou_token (nn.Embedding): Embedding for the IoU token. num_mask_tokens (int): Number of mask tokens. mask_tokens (nn.Embedding): Embedding for the mask tokens. output_upscaling (nn.Sequential): Neural network sequence for upscaling the output. output_hypernetworks_mlps (nn.ModuleList): Hypernetwork MLPs for generating masks. iou_prediction_head (nn.Module): MLP for predicting mask quality. )num_multimask_outputs activationiou_head_depthiou_head_hidden_dimtransformer_dim transformerr rrrreturnNc t|||_||_||_t j d||_|dz|_t j |j||_ t jt j||dzddt|dz|t j|dz|dzdd||_ t jt|jDcgc]}t!|||dzdc}|_t!|||j||_ycc}w)a Predicts masks given an image and prompt embeddings, using a transformer architecture. Args: transformer_dim (int): the channel dimension of the transformer module transformer (nn.Module): the transformer used to predict masks num_multimask_outputs (int): the number of masks to predict when disambiguating masks activation (nn.Module): the type of activation to use when upscaling masks iou_head_depth (int): the depth of the MLP used to predict mask quality iou_head_hidden_dim (int): the hidden dimension of the MLP used to predict mask quality ) kernel_sizestrider N)super__init__rrr r Embedding iou_tokennum_mask_tokens mask_tokens SequentialConvTranspose2droutput_upscaling ModuleListrangeMLPoutput_hypernetworks_mlpsiou_prediction_head) selfrrr rrr_ __class__s rC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\ultralytics/models/sam/modules/decoders.pyrzMaskDecoder.__init__s0* .&%:"a94q8<<(<(>#8#8$:J:J:Q:Q"RXYZ %//299:R:W:WXY:Z\^`bc M+CD!L%%&6 QQO++))(FLLOKYY 1a""38C1a7 Q1t';';#;