Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ViTAE-Transformer/MTP
/ functions
Functions
1,405 in github.com/ViTAE-Transformer/MTP
⨍
Functions
1,405
◇
Types & classes
310
↓ 1 callers
Function
get_mmcls_models
()
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/checkpoint.py:199
↓ 1 callers
Function
get_mmcls_models
()
Multi-Task_Pretrain/backbone/checkpoint.py:199
↓ 1 callers
Function
get_mmcls_models
()
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:199
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Change_Detection/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/custom_layer_decay_optimizer_constructor.py:20
↓ 1 callers
Function
get_num_layer_for_swin
(var_name, num_max_layer, depths)
Multi-Task_Pretrain/mmcv_custom/custom_layer_decay_optimizer_constructor.py:17
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Change_Detection/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/layer_decay_optimizer_constructor_vit.py:10
↓ 1 callers
Function
get_num_layer_for_vit
(var_name, num_max_layer)
Multi-Task_Pretrain/mmcv_custom/layer_decay_optimizer_constructor_vit.py:7
↓ 1 callers
Function
get_stage_id_for_convnext
Get the stage id to set the different learning rates in ``stage_wise`` decay_type. Args: var_name (str): The key of the model.
RS_Tasks_Finetune/Change_Detection/opencd/engine/optimizers/layer_decay_optimizer_constructor.py:55
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
Multi-Task_Pretrain/backbone/checkpoint.py:402
↓ 1 callers
Function
get_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:402
↓ 1 callers
Method
get_targets
Calculate the ground truth for all samples in a batch according to the sampling_results. Almost the same as the implementation in bbo
Multi-Task_Pretrain/rotated_detection/base_bbox_head.py:225
↓ 1 callers
Method
get_targets
Compute regression and classification targets for anchors in multiple images. Args: anchor_list (list[list[Tensor]]): Mul
Multi-Task_Pretrain/rotated_detection/anchor_head.py:311
↓ 1 callers
Method
get_targets
Calculate the ground truth for all samples in a batch according to the sampling_results. Almost the same as the implementation in bbo
Multi-Task_Pretrain/instance_segmentation/base_bbox_head.py:228
↓ 1 callers
Method
get_targets
Calculate the ground truth for all samples in a batch according to the sampling_results. Args: sampling_results (List[obj
Multi-Task_Pretrain/instance_segmentation/mask_head.py:153
↓ 1 callers
Method
gt_to_coco_json
Convert ground truth to coco format json file. Args: gt_dicts (Sequence[dict]): Ground truth of the dataset. outfile_
Multi-Task_Pretrain/instance_segmentation/metric.py:269
↓ 1 callers
Method
hue
Hue distortion. Args: img (np.ndarray): The input image. Returns: np.ndarray: Image after hue change.
Multi-Task_Pretrain/augmentations.py:1360
↓ 1 callers
Function
init_coco_format
()
scripts/prepare_xview_dataset.py:438
↓ 1 callers
Method
intersect_and_union
Calculate Intersection and Union. Args: pred_label (torch.tensor): Prediction segmentation map or predict result
Multi-Task_Pretrain/semantic_segmentation/metric.py:164
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load
(module, prefix='')
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:67
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_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:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/backbone/checkpoint.py:152
↓ 1 callers
Function
load_fileclient_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:152
↓ 1 callers
Function
load_img
Load image :param imgPath: path of the image to load :return: numpy array of the image
scripts/dota_submit_txt_to_fair1m_xml.py:9
↓ 1 callers
Function
load_img
Load image :param imgPath: path of the image to load :return: numpy array of the image
scripts/prepare_xview_dataset.py:170
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_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:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/backbone/checkpoint.py:124
↓ 1 callers
Function
load_pavimodel_dist
In distributed setting, this function only download checkpoint at local rank 0.
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:124
↓ 1 callers
Method
load_proposals
Load proposals from proposals file. The `proposals_list` should be a dict[img_path: proposals] with the same length as `data_list`. A
Multi-Task_Pretrain/datasets.py:1033
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Scene_Classification/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Semantic_Segmentation/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Horizontal_Detection/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Change_Detection/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmcv_custom/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
Multi-Task_Pretrain/backbone/checkpoint.py:41
↓ 1 callers
Function
load_state_dict
Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F
Multi-Task_Pretrain/mmcv_custom/checkpoint.py:41
↓ 1 callers
Method
loss
Calculate the loss based on the network predictions and targets. Args: cls_score (Tensor): Classification prediction
Multi-Task_Pretrain/rotated_detection/base_bbox_head.py:339
↓ 1 callers
Method
loss
Calculate losses from a batch of inputs and data samples.
Multi-Task_Pretrain/rotated_detection/base_detector.py:126
↓ 1 callers
Method
loss
Calculate the loss based on the network predictions and targets. Args: cls_score (Tensor): Classification prediction
Multi-Task_Pretrain/instance_segmentation/base_bbox_head.py:342
↓ 1 callers
Method
loss_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/instance_segmentation/rpn_head.py:99
↓ 1 callers
Function
main
()
scripts/convert_pkl_json.py:144
↓ 1 callers
Method
mask_loss
Perform forward propagation and loss calculation of the mask head on the features of the upstream network. Args: x (tuple
Multi-Task_Pretrain/rotated_detection/standard_roi_head.py:352
↓ 1 callers
Method
merge_det
Merging patch bboxes into full image. Args: results (list): Testing results of the dataset. nproc (int): number of pr
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/datasets/dota.py:216
↓ 1 callers
Method
merge_det
Merging patch bboxes into full image. Args: results (list): Testing results of the dataset. nproc (int): number of pr
RS_Tasks_Finetune/Rotated_Detection/mmrotate0.3.4/mmrotate/datasets/fair1m.py:224
↓ 1 callers
Method
merge_results
Merge patches' predictions into full image's results and generate a zip file for DOTA online evaluation. You can submit it at:
Multi-Task_Pretrain/rotated_detection/metric.py:473
↓ 1 callers
Function
mmengine_collate
(input_batch)
Multi-Task_Pretrain/utils.py:6
↓ 1 callers
Method
pad_gt_masks
Pad gt_masks to shape of batch_input_shape.
Multi-Task_Pretrain/preprocessing.py:223
↓ 1 callers
Method
pad_gt_sem_seg
Pad gt_sem_seg to shape of batch_input_shape.
Multi-Task_Pretrain/preprocessing.py:233
↓ 1 callers
Method
parse_data_info
Parse raw annotation to target format. Args: raw_data_info (dict): Raw data information load from ``ann_file`` Returns:
RS_Tasks_Finetune/Horizontal_Detection/mmdet/datasets/dior.py:92
↓ 1 callers
Method
parse_data_info
Parse raw annotation to target format. Args: raw_data_info (dict): Raw data information load from ``ann_file`` Returns:
RS_Tasks_Finetune/Horizontal_Detection/mmdet/datasets/xview.py:105
↓ 1 callers
Method
parse_data_info
Parse raw annotation to target format. Args: img_info (dict): Raw image information, usually it includes `img_id`
RS_Tasks_Finetune/Rotated_Detection/mmrotate1.x/mmrotate/datasets/dior.py:100
↓ 1 callers
Function
patch_replication_callback
Monkey-patch an existing `DataParallel` object. Add the replication callback. Useful when you have customized `DataParallel` implementation.
Multi-Task_Pretrain/sync_batchnorm/replicate.py:70
↓ 1 callers
Method
predict
Predict results from a batch of inputs and data samples with post- processing. Args: batch_inputs (Tensor): Inputs with s
Multi-Task_Pretrain/rotated_detection/two_stage.py:198
↓ 1 callers
Method
predict
Predict results from a batch of inputs and data samples with post- processing.
Multi-Task_Pretrain/rotated_detection/base_detector.py:132
↓ 1 callers
Method
predict
Perform forward propagation of the detection head and predict detection results on the features of the upstream network. Args:
Multi-Task_Pretrain/rotated_detection/dense_head.py:205
↓ 1 callers
Method
predict
Predict results from a batch of inputs and data samples with post- processing. Args: inputs (Tensor): Inputs with shape (
Multi-Task_Pretrain/semantic_segmentation/encoder_decoder.py:197
↓ 1 callers
Method
predict
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/two_stage.py:196
↓ 1 callers
Method
predict
Perform forward propagation of the detection head and predict detection results on the features of the upstream network. Args:
Multi-Task_Pretrain/instance_segmentation/rpn_head.py:362
↓ 1 callers
Method
predict_bbox
Perform forward propagation of the bbox head and predict detection results on the features of the upstream network. Args:
Multi-Task_Pretrain/rotated_detection/standard_roi_head.py:442
↓ 1 callers
Method
predict_mask
Perform forward propagation of the mask head and predict detection results on the features of the upstream network. Args:
Multi-Task_Pretrain/rotated_detection/standard_roi_head.py:519
↓ 1 callers
Function
print_map_summary
Print mAP and results of each class. A table will be printed to show the gts/dets/recall/AP of each class and the mAP. Args: mea
Multi-Task_Pretrain/rotated_detection/metric.py:27
↓ 1 callers
Method
psp_forward
Forward function of PSP module.
RS_Tasks_Finetune/Change_Detection/opencd/models/decode_heads/uper_head.py:76
↓ 1 callers
Method
register_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
Multi-Task_Pretrain/sync_batchnorm/comm.py:84
↓ 1 callers
Method
regress_by_class
Regress the bbox for the predicted class. Used in Cascade R-CNN. Args: priors (Tensor): Priors from `rpn_head` or last stage
Multi-Task_Pretrain/rotated_detection/base_bbox_head.py:696
↓ 1 callers
Method
regress_by_class
Regress the bbox for the predicted class. Used in Cascade R-CNN. Args: priors (Tensor): Priors from `rpn_head` or last stage
Multi-Task_Pretrain/instance_segmentation/base_bbox_head.py:700
↓ 1 callers
Function
remove_center_sampling_locations
(sampling_locations, kernel_w, kernel_h)
RS_Tasks_Finetune/Scene_Classification/mmpretrain/models/backbones/ops_dcnv3/functions/dcnv3_func.py:167
↓ 1 callers
Function
remove_center_sampling_locations
(sampling_locations, kernel_w, kernel_h)
RS_Tasks_Finetune/Semantic_Segmentation/mmseg/models/backbones/ops_dcnv3/functions/dcnv3_func.py:167
↓ 1 callers
Function
remove_center_sampling_locations
(sampling_locations, kernel_w, kernel_h)
RS_Tasks_Finetune/Horizontal_Detection/mmdet/models/backbones/ops_dcnv3/functions/dcnv3_func.py:167
← previous
next →
401–500 of 1,405, ranked by callers