MCPcopy Create free account

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

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

⨍ Functions 528◇ Types & classes 132↳ Endpoints 1
Method__call__
(self, results)
segmentation/mmseg_custom/datasets/pipelines/formatting.py:58
Method__call__
Args: cls_pred (Tensor): Predicted classification logits, shape [num_query, num_class]. gt_labels (Te
segmentation/mmseg_custom/models/losses/match_loss.py:38
Method__call__
Args: cls_pred (Tensor): Predicted classfication logits in shape (N1, H, W), dtype=torch.float32. gt_
segmentation/mmseg_custom/models/losses/match_loss.py:67
Method__call__
Args: cls_pred (Tensor): Predicted classification logits, shape [num_query, num_class]. gt_labels (Te
segmentation/mmseg_custom/models/losses/match_loss.py:115
Method__call__
Args: mask_preds (Tensor): Mask prediction logits in shape (N1, H, W). gt_masks (Tensor): Ground truth in shape (N2,
segmentation/mmseg_custom/models/losses/match_loss.py:167
Method__call__
Args: cls_pred (Tensor): Predicted classification logits, shape [num_query, num_class]. gt_labels (Te
segmentation/mmseg_custom/models/losses/match_costs.py:38
Method__call__
Args: cls_pred (Tensor): Predicted classfication logits in shape (N1, H, W), dtype=torch.float32. gt_
segmentation/mmseg_custom/models/losses/match_costs.py:67
Method__call__
Args: cls_pred (Tensor): Predicted classification logits, shape [num_query, num_class]. gt_labels (Te
segmentation/mmseg_custom/models/losses/match_costs.py:115
Method__call__
Args: mask_preds (Tensor): Mask prediction logits in shape (N1, H, W). gt_masks (Tensor): Ground truth in shape (N2,
segmentation/mmseg_custom/models/losses/match_costs.py:167
Method__call__
Args: cls_pred (Tensor): Predicted classification logits. gt_labels (Tensor): Labels. Returns: Te
segmentation/mmseg_custom/models/losses/match_costs.py:219
Method__init__
Multi-Scale Deformable Attention Module. :param d_model hidden dimension :param n_levels number of feature levels :p
detection/ops/modules/ms_deform_attn.py:29
Method__init__
(self, in_channels, out_channels, kernel_size=1,
detection/mmdet_custom/models/necks/channel_mapper.py:43
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
detection/mmdet_custom/models/necks/extra_attention.py:33
Method__init__
(self, in_channels, num_head, with_ffn=True,
detection/mmdet_custom/models/necks/extra_attention.py:62
Method__init__
(self, pretrain_size=224, out_indices=None, *args, **kwargs)
detection/mmdet_custom/models/backbones/vit_baseline.py:16
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
detection/mmdet_custom/models/backbones/comer_modules.py:67
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
detection/mmdet_custom/models/backbones/comer_modules.py:89
Method__init__
(self, dim=768)
detection/mmdet_custom/models/backbones/comer_modules.py:111
Method__init__
(self, dim=768)
detection/mmdet_custom/models/backbones/comer_modules.py:129
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
detection/mmdet_custom/models/backbones/comer_modules.py:182
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
detection/mmdet_custom/models/backbones/comer_modules.py:221
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
detection/mmdet_custom/models/backbones/comer_modules.py:267
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, norm_layer=partial(nn.Laye
detection/mmdet_custom/models/backbones/comer_modules.py:322
Method__init__
(self, dim, num_heads=6, n_points=4, norm_layer=partial(nn.LayerNorm, eps=1e-6), drop=0., dro
detection/mmdet_custom/models/backbones/comer_modules.py:362
Method__init__
(self, pretrain_size=224, num_heads=12, conv_inplane=64, n_points=4, deform_num_heads=6, init
detection/mmdet_custom/models/backbones/vit_comer.py:20
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True,
detection/mmdet_custom/models/backbones/base/vit.py:41
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
detection/mmdet_custom/models/backbones/base/vit.py:66
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0., window_size=14)
detection/mmdet_custom/models/backbones/base/vit.py:124
Method__init__
(self, normalized_shape, eps=1e-6)
detection/mmdet_custom/models/backbones/base/vit.py:218
Method__init__
Args: in_channels (int): Number of input channels. out_channels (int): Number of output channels. bottlen
detection/mmdet_custom/models/backbones/base/vit.py:239
Method__init__
Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number
detection/mmdet_custom/models/backbones/base/vit.py:363
Method__init__
(self, drop_prob=None)
detection/mmdet_custom/models/backbones/base/beit.py:62
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., window_si
detection/mmdet_custom/models/backbones/base/beit.py:95
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
detection/mmdet_custom/models/backbones/base/beit.py:199
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
detection/mmdet_custom/models/backbones/base/beit.py:245
Method__init__
(self, backbone, img_size=224, feature_size=None, in_chans=3, embed_dim=768)
detection/mmdet_custom/models/backbones/base/beit.py:273
Method__init__
(self, window_size, num_heads)
detection/mmdet_custom/models/backbones/base/beit.py:305
Method__init__
(self, img_size=512, patch_size=16, in_chans=3, num_classes=80, embed_dim=768, depth=12, num_
detection/mmdet_custom/models/backbones/base/beit.py:340
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0., window_size=14)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:71
Method__init__
(self, hidden_size=768, intermediate_size=3072, num_attention_heads=12, drop_path_ratio=0.1,
detection/mmdet_custom/models/backbones/base/uniperceiver.py:110
Method__init__
(self, in_dim=3, out_dim=768, patch_size=16, image_size=224, dropout=0.)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:154
Method__init__
(self, img_size=(224, 224), patch_size=(16, 16), in_chans=3, embed_dim=768)
detection/mmdet_custom/models/backbones/base/uniperceiver.py:187
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, depth=12, num_heads=12, mlp_ra
detection/mmdet_custom/models/backbones/base/uniperceiver.py:218
Method__init__
(self, **kwargs)
detection/mmdet_custom/models/detectors/htc_aug.py:15
Method__init__
(self, *args, **kwargs)
segmentation/mmseg_custom/core/utils/dist_utils.py:60
Method__init__
(self, strides, offset=0.5)
segmentation/mmseg_custom/core/anchor/point_generator.py:53
Method__init__
(self, num, pos_fraction, neg_pos_ub=-1, a
segmentation/mmseg_custom/core/box/samplers/base_sampler.py:11
Method__init__
(self, pos_inds, neg_inds, masks, gt_masks, assign_result, gt_flags)
segmentation/mmseg_custom/core/box/samplers/mask_sampling_result.py:12
Method__init__
(self, pos_inds, neg_inds, bboxes, gt_bboxes, assign_result, gt_flags)
segmentation/mmseg_custom/core/box/samplers/sampling_result.py:24
Method__init__
(self, **kwargs)
segmentation/mmseg_custom/core/box/samplers/mask_pseudo_sampler.py:15
Method__init__
(self, **kwargs)
segmentation/mmseg_custom/datasets/potsdam.py:19
Method__init__
(self, **kwargs)
segmentation/mmseg_custom/datasets/mapillary.py:41
Method__init__
(self, img_scale=None, multiscale_mode='range', ratio_range
segmentation/mmseg_custom/datasets/pipelines/transform.py:34
Method__init__
(self, size=None, pad_val=0, seg_pad_val=255)
segmentation/mmseg_custom/datasets/pipelines/transform.py:260
Method__init__
(self)
segmentation/mmseg_custom/datasets/pipelines/transform.py:313
Method__init__
(self, ignore_index=255)
segmentation/mmseg_custom/datasets/pipelines/formatting.py:55
Method__init__
(self, backbone, decode_head, neck=None, a
segmentation/mmseg_custom/models/segmentors/encoder_decoder_mask2former.py:20
Method__init__
(self, backbone, decode_head, neck=None, a
segmentation/mmseg_custom/models/segmentors/encoder_decoder_mask2former_aug.py:20
Method__init__
(self, pretrain_size=224, *args, **kwargs)
segmentation/mmseg_custom/models/backbones/vit_baseline.py:18
Method__init__
(self, drop_prob=None)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:29
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., window_si
segmentation/mmseg_custom/models/backbones/beit_baseline.py:62
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
segmentation/mmseg_custom/models/backbones/beit_baseline.py:152
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:186
Method__init__
(self, backbone, img_size=224, feature_size=None, in_chans=3, embed_dim=768)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:215
Method__init__
(self, window_size, num_heads)
segmentation/mmseg_custom/models/backbones/beit_baseline.py:248
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, num_classes=80, embed_dim=768, depth=12, num_
segmentation/mmseg_custom/models/backbones/beit_baseline.py:290
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
segmentation/mmseg_custom/models/backbones/comer_modules.py:67
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
segmentation/mmseg_custom/models/backbones/comer_modules.py:89
Method__init__
(self, dim=768)
segmentation/mmseg_custom/models/backbones/comer_modules.py:111
Method__init__
(self, dim=768)
segmentation/mmseg_custom/models/backbones/comer_modules.py:129
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
segmentation/mmseg_custom/models/backbones/comer_modules.py:182
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
segmentation/mmseg_custom/models/backbones/comer_modules.py:221
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, with_cffn=True, cffn_ratio
segmentation/mmseg_custom/models/backbones/comer_modules.py:274
Method__init__
(self, dim, num_heads=6, n_points=4, n_levels=1, deform_ratio=1.0, norm_layer=partial(nn.Laye
segmentation/mmseg_custom/models/backbones/comer_modules.py:329
Method__init__
(self, dim, num_heads=6, n_points=4, norm_layer=partial(nn.LayerNorm, eps=1e-6), drop=0., dro
segmentation/mmseg_custom/models/backbones/comer_modules.py:370
Method__init__
(self, pretrain_size=224, num_heads=12, conv_inplane=64, n_points=4, deform_num_heads=6, init
segmentation/mmseg_custom/models/backbones/vit_comer.py:20
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True)
segmentation/mmseg_custom/models/backbones/base/vit.py:41
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
segmentation/mmseg_custom/models/backbones/base/vit.py:65
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0., window_size=14, pad_mode
segmentation/mmseg_custom/models/backbones/base/vit.py:122
Method__init__
Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number
segmentation/mmseg_custom/models/backbones/base/vit.py:260
Method__init__
(self, drop_prob=None)
segmentation/mmseg_custom/models/backbones/base/beit.py:29
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., window_si
segmentation/mmseg_custom/models/backbones/base/beit.py:62
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
segmentation/mmseg_custom/models/backbones/base/beit.py:152
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
segmentation/mmseg_custom/models/backbones/base/beit.py:194
Method__init__
(self, backbone, img_size=224, feature_size=None, in_chans=3, embed_dim=768)
segmentation/mmseg_custom/models/backbones/base/beit.py:223
Method__init__
(self, window_size, num_heads)
segmentation/mmseg_custom/models/backbones/base/beit.py:256
Method__init__
(self, img_size=512, patch_size=16, in_chans=3, num_classes=80, embed_dim=768, depth=12, num_
segmentation/mmseg_custom/models/backbones/base/beit.py:298
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0., window_size=14)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:71
Method__init__
(self, hidden_size=768, intermediate_size=3072, num_attention_heads=12, drop_path_ratio=0.1,
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:110
Method__init__
(self, in_dim=3, out_dim=768, patch_size=16, image_size=224, dropout=0.)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:154
Method__init__
(self, img_size=(224, 224), patch_size=(16, 16), in_chans=3, embed_dim=768)
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:188
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, depth=12, num_heads=12, mlp_ra
segmentation/mmseg_custom/models/backbones/base/uniperceiver.py:219
Method__init__
(self, num_feats, row_num_embed=50, col_num_embed=50,
segmentation/mmseg_custom/models/utils/positional_encoding.py:119
Method__init__
(self, kernel_size=1, stride=1, dilation=1, padding='corner')
segmentation/mmseg_custom/models/utils/transformer.py:67
Method__init__
(self, in_channels, out_channels, kernel_size=2,
segmentation/mmseg_custom/models/utils/transformer.py:138
Method__init__
(self, attn_cfgs, feedforward_channels, ffn_dropout=0.0,
segmentation/mmseg_custom/models/utils/transformer.py:357
Method__init__
(self, *args, post_norm_cfg=dict(type='LN'), **kwargs)
segmentation/mmseg_custom/models/utils/transformer.py:388
Method__init__
(self, *args, post_norm_cfg=dict(type='LN'), return_interme
segmentation/mmseg_custom/models/utils/transformer.py:420
Method__init__
(self, encoder=None, decoder=None, init_cfg=None)
segmentation/mmseg_custom/models/utils/transformer.py:485
Method__init__
(self, *args, return_intermediate=False, **kwargs)
segmentation/mmseg_custom/models/utils/transformer.py:557
← previousnext →201–300 of 528, ranked by callers