MCPcopy Create free account

hub / github.com/ViTAE-Transformer/MTP / functions

Functions1,405 in github.com/ViTAE-Transformer/MTP

↓ 261 callersMethodget
(self)
Multi-Task_Pretrain/sync_batchnorm/comm.py:32
↓ 17 callersFunctionmain_process
(args)
Multi-Task_Pretrain/main_pretrain.py:104
↓ 13 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/intern_image.py:41
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/intern_image.py:41
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/intern_image.py:41
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/intern_image.py:41
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/intern_image.py:41
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/intern_image.py:43
↓ 12 callersFunctionbuild_norm_layer
(dim, norm_layer, in_format='channels_last', ou
Multi-Task_Pretrain/backbone/intern_image.py:39
↓ 11 callersMethodevaluate
Evaluate the dataset. Args: results (list): Testing results of the dataset. metric (str | list[str]): Metrics to be e
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/datasets/dota.py:165
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/intern_image.py:315
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/intern_image.py:315
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/intern_image.py:315
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/intern_image.py:315
↓ 9 callersMethod__init__
(self, min=0, max=1)
RS_Tasks_Finetune/Change_Detection/opencd/models/decode_heads/unet_head.py:89
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/intern_image.py:315
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/intern_image.py:317
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None
Multi-Task_Pretrain/backbone/intern_image.py:313
↓ 9 callersMethodprocess
Process one batch of data samples and predictions. The processed results should be stored in ``self.results``, which will be used to c
Multi-Task_Pretrain/rotated_detection/metric.py:621
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/vit_rvsa_mtp.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/vit_rvsa_mtp.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/vit_rvsa_mtp_branches.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/vit_rvsa_mtp.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/vit_rvsa_mtp.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/vit_rvsa_mtp_branches.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/vit_rvsa_mtp.py:36
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/vit_rvsa_mtp_branches.py:38
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/vit_rvsa_mtp.py:38
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
Multi-Task_Pretrain/backbone/vit_win_rvsa_v3_wsz7.py:46
↓ 8 callersMethod_load_from_state_dict
Exchange bbox_head key to rpn_head key when loading single-stage weights into two-stage model.
Multi-Task_Pretrain/rotated_detection/two_stage.py:68
↓ 8 callersMethodadd_pred_to_datasample
Add predictions to `DetDataSample`. Args: data_samples (list[:obj:`DetDataSample`], optional): A batch of data sa
Multi-Task_Pretrain/rotated_detection/base_detector.py:154
↓ 6 callersMethodbackward
(ctx, grad_output)
Multi-Task_Pretrain/backbone/ops_dcnv3/functions/dcnv3_func.py:61
↓ 6 callersFunctiondata_augs
(dataset, datainfos)
Multi-Task_Pretrain/utils.py:104
↓ 6 callersMethodgen_sampling_results
(self, tbr)
Multi-Task_Pretrain/instance_segmentation/roi_head.py:181
↓ 6 callersMethodget_data_info
Get annotation by index and automatically call ``full_init`` if the dataset has not been fully initialized. Args: idx (in
Multi-Task_Pretrain/datasets.py:350
↓ 6 callersFunctionparse_datainfos
(dataset, img_ids)
Multi-Task_Pretrain/utils.py:74
↓ 6 callersMethodtest_bbox_generate_roi
(self, rpn_results_list)
Multi-Task_Pretrain/instance_segmentation/roi_head.py:238
↓ 6 callersMethodtest_before_roihead
Predict results from a batch of inputs and data samples with post- processing. Args: batch_inputs (Tensor): Inputs with s
Multi-Task_Pretrain/instance_segmentation/mask_rcnn.py:189
↓ 6 callersMethodtrain_before_roihead
Calculate losses from a batch of inputs and data samples. Args: batch_inputs (Tensor): Input images of shape (N, C, H, W).
Multi-Task_Pretrain/instance_segmentation/mask_rcnn.py:137
↓ 5 callersFunction_unsqueeze_ft
add new dimensions at the front and the tail
Multi-Task_Pretrain/sync_batchnorm/batchnorm.py:52
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctiondcnv3_core_pytorch
( input, offset, mask, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation_
Multi-Task_Pretrain/backbone/ops_dcnv3/functions/dcnv3_func.py:174
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/backbone/checkpoint.py:110
↓ 5 callersFunctionload_url_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:110
↓ 5 callersMethodpredict_by_feat
Transform a batch of output features extracted from the head into bbox results. Args: rois (tuple[Tensor]): Tuple of boxe
Multi-Task_Pretrain/rotated_detection/base_bbox_head.py:438
↓ 4 callersMethod_forward_feature
Forward function for feature maps before classifying each pixel with ``self.cls_seg`` fc. Args: inputs (list[Tensor]): Li
RS_Tasks_Finetune/Change_Detection/opencd/models/decode_heads/uper_head.py:86
↓ 4 callersMethod_get_unserialized_subset
Get subset of data information list. Args: indices (int or Sequence[int]): If type of indices is int, indices rep
Multi-Task_Pretrain/datasets.py:831
↓ 4 callersMethod_mask_forward
Mask head forward function used in both training and testing. Args: x (tuple[Tensor]): Tuple of multi-level img features.
Multi-Task_Pretrain/rotated_detection/standard_roi_head.py:406
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodbackward
(ctx, grad_output)
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/ops_dcnv3/functions/dcnv3_func.py:61
↓ 4 callersMethodpredict_by_feat
Transform a batch of output features extracted from the head into mask results. Args: mask_preds (tuple[Tensor]): Tuple o
Multi-Task_Pretrain/instance_segmentation/mask_head.py:219
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Change_Detection/opencd/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/models/backbones/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod__init__
DCNv3 Module :param channels :param kernel_size :param stride :param pad :param dilation :par
Multi-Task_Pretrain/backbone/ops_dcnv3/modules/dcnv3.py:222
↓ 3 callersMethod_add_conv_fc_branch
Add shared or separable branch. convs -> avg pool (optional) -> fcs
Multi-Task_Pretrain/rotated_detection/bbox_head.py:129
↓ 3 callersMethod_add_conv_fc_branch
Add shared or separable branch. convs -> avg pool (optional) -> fcs
Multi-Task_Pretrain/instance_segmentation/bbox_head.py:128
↓ 3 callersMethodassign
Assign gt to bboxes. This method assign a gt bbox to every bbox (proposal/anchor), each bbox will be assigned with -1, or a semi-posi
Multi-Task_Pretrain/rotated_detection/max_iou_assigner.py:139
↓ 3 callersMethodbbox_loss
Perform forward propagation and loss calculation of the bbox head on the features of the upstream network. Args: x (tuple
Multi-Task_Pretrain/instance_segmentation/roi_head.py:281
↓ 3 callersMethodconvert
Multiple with alpha and add beat with clip. Args: img (np.ndarray): The input image. alpha (int): Image weights, chan
Multi-Task_Pretrain/augmentations.py:1290
↓ 3 callersFunctioneval_rbbox_map
Evaluate mAP of a rotated dataset. Args: det_results (list[list]): [[cls1_det, cls2_det, ...], ...]. The outer list indicates
Multi-Task_Pretrain/rotated_detection/metric.py:236
↓ 3 callersMethodextract_feat
Extract features. Args: batch_inputs (Tensor): Image tensor with shape (N, C, H ,W). Returns: tuple[Tensor]:
Multi-Task_Pretrain/rotated_detection/two_stage.py:102
↓ 3 callersMethodextract_feat
Extract features from images.
Multi-Task_Pretrain/semantic_segmentation/encoder_decoder.py:126
↓ 3 callersMethodextract_feat
Extract features. Args: batch_inputs (Tensor): Image tensor with shape (N, C, H ,W). Returns: tuple[Tensor]:
Multi-Task_Pretrain/instance_segmentation/two_stage.py:100
↓ 3 callersMethodforward
(self, x)
Multi-Task_Pretrain/backbone/intern_image.py:327
↓ 3 callersMethodinference
Inference with slide/whole style. Args: inputs (Tensor): The input image of shape (N, 3, H, W). batch_img_metas (List
Multi-Task_Pretrain/semantic_segmentation/encoder_decoder.py:334
↓ 3 callersMethodinit_weights
(self, pretrained)
Multi-Task_Pretrain/backbone/intern_image.py:641
↓ 3 callersFunctionload
(module, prefix='')
Multi-Task_Pretrain/backbone/checkpoint.py:67
↓ 3 callersMethodloss
Calculate losses from a batch of inputs and data samples. Args: inputs (Tensor): Input images. data_samples (list[:ob
Multi-Task_Pretrain/semantic_segmentation/encoder_decoder.py:171
↓ 3 callersMethodloss_by_feat
Calculate the loss based on the features extracted by the detection head. Args: cls_scores (list[Tensor]): Box scores for
Multi-Task_Pretrain/rotated_detection/rpn_head.py:99
↓ 3 callersFunctionparse_losses
Parses the raw outputs (losses) of the network. Args: losses (dict): Raw output of the network, which usually contain losses
Multi-Task_Pretrain/utils.py:140
↓ 3 callersMethodpredict_bbox
Perform forward propagation of the bbox head and predict detection results on the features of the upstream network. Args:
Multi-Task_Pretrain/instance_segmentation/roi_head.py:488
↓ 3 callersMethodput
(self, result)
Multi-Task_Pretrain/sync_batchnorm/comm.py:26
↓ 3 callersMethodsample
Sample positive and negative bboxes. This is a simple implementation of bbox sampling given candidates, assigning results and ground
Multi-Task_Pretrain/rotated_detection/random_sampler.py:49
↓ 3 callersMethodtest_insseg_roi_head_box_head_forward
(self, n, batch_img_metas, proposals, rois, rcnn_test_cfg, num_classes, bbox_rescale, fc_cls=None, fc_reg=None
Multi-Task_Pretrain/models.py:271
↓ 3 callersMethodtest_insseg_roi_head_mask_head_forward
(self, n, batch_img_metas, mask_rois, results_list, rescale, conv_logits=None)
Multi-Task_Pretrain/models.py:295
↓ 3 callersMethodtest_mask_generate_rois
(self, results_list)
Multi-Task_Pretrain/instance_segmentation/roi_head.py:427
↓ 3 callersMethodtest_rotdet_roi_head_box_head_forward
(self, r, batch_img_metas, proposals, rois, rcnn_test_cfg, num_classes, bbox_rescale, fc_cls=None, fc_reg=None
Multi-Task_Pretrain/models.py:240
↓ 3 callersMethodtrain_insseg_roi_head_box_head_forward
(self, n, sampling_results, num_classes, fc_cls=None, fc_reg=None)
Multi-Task_Pretrain/models.py:253
↓ 3 callersMethodtrain_insseg_roi_head_mask_head_forward
(self, n, sampling_results, bbox_feats, batch_gt_instance, data_sample, conv_logits=None)
Multi-Task_Pretrain/models.py:284
↓ 3 callersMethodtrain_rotdet_roi_head_box_head_forward
(self, r, sampling_results, num_classes, fc_cls=None, fc_reg=None)
Multi-Task_Pretrain/models.py:224
↓ 2 callersMethod__init__
(self, transforms: Optional[Sequence[Union[dict, Callable]]])
Multi-Task_Pretrain/datasets.py:28
next →1–100 of 1,405, ranked by callers