MCPcopy Create free account

hub / github.com/Traffic-X/ViT-CoMer / functions

Functions528 in github.com/Traffic-X/ViT-CoMer

Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/vit.py:56
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/vit.py:78
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/vit.py:137
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/vit.py:225
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/vit.py:281
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/vit.py:328
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/vit.py:448
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/beit.py:66
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/beit.py:84
Methodforward
(self, x, H, W, rel_pos_bias=None)
detection/mmdet_custom/models/backbones/base/beit.py:140
Methodforward
(self, x, H, W, rel_pos_bias=None)
detection/mmdet_custom/models/backbones/base/beit.py:224
Methodforward
(self, x, **kwargs)
detection/mmdet_custom/models/backbones/base/beit.py:258
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/beit.py:297
Methodforward
(self)
detection/mmdet_custom/models/backbones/base/beit.py:330
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:55
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:83
Methodforward
(self, x, H, W)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:138
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:167
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:206
Methodforward
(self, x)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:252
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/vit_baseline.py:86
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:33
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:51
Methodforward
(self, x, rel_pos_bias=None)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:117
Methodforward
(self, x, rel_pos_bias=None)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:172
Methodforward
(self, x, **kwargs)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:198
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:239
Methodforward
(self)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:277
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:421
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:78
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:100
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:115
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:152
Methodforward
(self, query, reference_points, feat, spatial_shapes, level_start_index, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:198
Methodforward
(self, query, reference_points, feat, spatial_shapes, level_start_index, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:243
Methodforward
(self, query, reference_points, feat, spatial_shapes, level_start_index, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:296
Methodforward
(self, query, reference_points, feat, spatial_shapes, level_start_index, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:344
Methodforward
(self, x, c, blocks, deform_inputs1, deform_inputs2, H, W)
segmentation/mmseg_custom/models/backbones/comer_modules.py:407
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/comer_modules.py:471
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/vit_comer.py:100
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/vit.py:55
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/vit.py:76
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/vit.py:136
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/vit.py:232
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/vit.py:334
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/beit.py:33
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/beit.py:51
Methodforward
(self, x, rel_pos_bias=None)
segmentation/mmseg_custom/models/backbones/base/beit.py:116
Methodforward
(self, x, H, W, rel_pos_bias=None)
segmentation/mmseg_custom/models/backbones/base/beit.py:173
Methodforward
(self, x, **kwargs)
segmentation/mmseg_custom/models/backbones/base/beit.py:206
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/beit.py:247
Methodforward
(self)
segmentation/mmseg_custom/models/backbones/base/beit.py:285
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:55
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:83
Methodforward
(self, x, H, W)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:138
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:167
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:207
Methodforward
(self, x)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:253
Methodforward
Forward function for `SinePositionalEncoding`. Args: mask (Tensor): ByteTensor mask. Non-zero values representing
segmentation/mmseg_custom/models/utils/positional_encoding.py:55
Methodforward
Forward function for `LearnedPositionalEncoding`. Args: mask (Tensor): ByteTensor mask. Non-zero values representing
segmentation/mmseg_custom/models/utils/positional_encoding.py:131
Methodforward
(self, x)
segmentation/mmseg_custom/models/utils/transformer.py:95
Methodforward
Args: x (Tensor): Has shape (B, H*W, C_in). input_size (tuple[int]): The spatial shape of x, arrange as (H, W).
segmentation/mmseg_custom/models/utils/transformer.py:187
Methodforward
Forward function for `TransformerCoder`. Returns: Tensor: forwarded results with shape [num_query, bs, embed_dims].
segmentation/mmseg_custom/models/utils/transformer.py:399
Methodforward
Forward function for `TransformerDecoder`. Args: query (Tensor): Input query with shape `(num_query, bs, embed_di
segmentation/mmseg_custom/models/utils/transformer.py:434
Methodforward
Forward function for `Transformer`. Args: x (Tensor): Input query with shape [bs, c, h, w] where c = embed_dims.
segmentation/mmseg_custom/models/utils/transformer.py:498
Methodforward
Forward function for `TransformerDecoder`. Args: query (Tensor): Input query with shape `(num_query, bs, embed_di
segmentation/mmseg_custom/models/utils/transformer.py:562
Methodforward
Forward function for `Transformer`. Args: mlvl_feats (list(Tensor)): Input queries from different level. Each ele
segmentation/mmseg_custom/models/utils/transformer.py:818
Methodforward
Forward function for `DynamicConv`. Args: param_feature (Tensor): The feature can be used to generate the paramet
segmentation/mmseg_custom/models/utils/transformer.py:1040
Methodforward
Forward function. Args: feats (list[Tensor]): Multi scale Features from the upstream network, each is a 4D-tensor
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py:446
Methodforward
Forward function. Args: feats (list[Tensor]): Features from the upstream network, each is a 4D-tensor.
segmentation/mmseg_custom/models/decode_heads/maskformer_head.py:398
Methodforward
Args: feats (list[Tensor]): Feature maps of each level. Each has shape of (batch_size, c, h, w). Returns
segmentation/mmseg_custom/models/plugins/msdeformattn_pixel_decoder.py:160
Methodforward
Args: feats (list[Tensor]): Feature maps of each level. Each has shape of [bs, c, h, w]. img_metas (l
segmentation/mmseg_custom/models/plugins/pixel_decoder.py:85
Methodforward
Args: feats (list[Tensor]): Feature maps of each level. Each has shape of [bs, c, h, w]. img_metas (l
segmentation/mmseg_custom/models/plugins/pixel_decoder.py:185
Methodforward
Forward function. Args: pred (torch.Tensor): The prediction, has a shape (n, *). target (torch.Tensor): The label of
segmentation/mmseg_custom/models/losses/dice_loss.py:128
Methodforward
Forward function.
segmentation/mmseg_custom/models/losses/cross_entropy_loss.py:249
Methodforward
Forward function. Args: pred (torch.Tensor): The prediction. target (torch.Tensor): The learning label of the predict
segmentation/mmseg_custom/models/losses/focal_loss.py:135
Methodforward_dummy
Dummy forward function.
segmentation/mmseg_custom/models/segmentors/encoder_decoder_mask2former.py:116
Methodforward_dummy
Dummy forward function.
segmentation/mmseg_custom/models/segmentors/encoder_decoder_mask2former_aug.py:116
Methodforward_test
Test segment without test-time aumengtation. Only the output of last decoder layers was used. Args: inputs (list[Tensor]
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py:557
Methodforward_train
Forward function for training mode. Args: x (list[Tensor]): Multi-level features from the upstream network, each
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py:527
Methodforward_train
Forward function for training mode. Args: x (list[Tensor]): Multi-level features from the upstream network, each
segmentation/mmseg_custom/models/decode_heads/maskformer_head.py:463
Methodget_num_layers
(self)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:390
Methodgrid_points
(self, featmap_size, stride=16, device='cuda')
segmentation/mmseg_custom/core/anchor/point_generator.py:19
Methodgrid_priors
Generate grid points of multiple feature levels. Args: featmap_sizes (list[tuple]): List of feature map sizes in
segmentation/mmseg_custom/core/anchor/point_generator.py:78
Methodinfo
Returns a dictionary of info about the object.
segmentation/mmseg_custom/core/box/samplers/mask_sampling_result.py:49
Methodinfo
Returns a dictionary of info about the object.
segmentation/mmseg_custom/core/box/samplers/sampling_result.py:79
Methodinit_weights
(self, pretrained=None)
segmentation/mmseg_custom/models/backbones/vit_baseline.py:48
Methodinit_weights
Initialize the transformer weights.
segmentation/mmseg_custom/models/utils/transformer.py:676
Methodinit_weights
(self)
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py:143
Methodinit_weights
Initialize weights.
segmentation/mmseg_custom/models/plugins/msdeformattn_pixel_decoder.py:134
Methodinit_weights
Initialize weights.
segmentation/mmseg_custom/models/plugins/pixel_decoder.py:76
Methodinit_weights
Initialize weights.
segmentation/mmseg_custom/models/plugins/pixel_decoder.py:175
Methodlog
(self, runner)
detection/mmcv_custom/customized_text.py:89
Methodloss_name
Loss Name. This function must be implemented and will return the name of this loss function. This name will be used to combine differ
segmentation/mmseg_custom/models/losses/cross_entropy_loss.py:279
Methodloss_single
Loss function for outputs from a single decoder layer. Args: cls_scores (Tensor): Mask score logits from a single decoder layer
segmentation/mmseg_custom/models/decode_heads/mask2former_head.py:269
Methodloss_single
Loss function for outputs from a single decoder layer. Args: cls_scores (Tensor): Mask score logits from a single decoder layer
segmentation/mmseg_custom/models/decode_heads/maskformer_head.py:302
Functionmask2bbox
Obtain tight bounding boxes of binary masks. Args: masks (Tensor): Binary mask of shape (n, h, w). Returns: Tensor: Bboxe wi
segmentation/mmseg_custom/core/mask/utils.py:68
Functionmask_cross_entropy
Calculate the CrossEntropy loss for masks. Args: pred (torch.Tensor): The prediction with shape (N, C), C is the number of cl
segmentation/mmseg_custom/models/losses/cross_entropy_loss.py:153
Methodmasks
torch.Tensor: concatenated positive and negative boxes
segmentation/mmseg_custom/core/box/samplers/mask_sampling_result.py:36
Functionms_deform_attn_backward
detection/ops/src/ms_deform_attn.h:41
← previousnext →401–500 of 528, ranked by callers