MCPcopy Create free account

hub / github.com/OpenGVLab/unmasked_teacher / functions

Functions1,198 in github.com/OpenGVLab/unmasked_teacher

↓ 1 callersMethod_video_TSN_decord_batch_loader
(self, directory, video_reader, duration, indices, skip_offsets)
single_modality/datasets/mae.py:262
↓ 1 callersFunctionadd_different_lr
use lr=diff_lr for modules named found in diff_lr_names, otherwise use lr=default_lr Args: named_param_tuples_or_model: List([name, p
multi_modality/utils/optimizer.py:31
↓ 1 callersFunctionadd_weight_decay
(model, weight_decay, no_decay_list=(), filter_bias_and_bn=True)
multi_modality/utils/optimizer.py:17
↓ 1 callersFunctionalign_and_update_state_dicts
Strategy: suppose that the models that we will create will have prefixes appended to each of its keys, for example due to an extra level of n
single_modality/action_detection/alphaction/utils/model_serialization.py:8
↓ 1 callersFunctionall_reduce
(tensor, average=False)
single_modality/action_detection/alphaction/utils/comm.py:201
↓ 1 callersFunctionarea
Computes area of boxes. Args: boxlist: BoxList holding N boxes Returns: a numpy array with shape [N*1] representing box areas
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list_ops.py:38
↓ 1 callersMethodarea
(self)
single_modality/action_detection/alphaction/structures/bounding_box.py:312
↓ 1 callersFunctionava_evaluation
(dataset, predictions, output_folder, **_)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/__init__.py:5
↓ 1 callersFunctionbrightness_jitter
Perfrom brightness jittering on the input images. The channels of images should be in order BGR. Args: var (float): jitter ratio
single_modality/datasets/video_transforms.py:348
↓ 1 callersFunctionbuild_3d_roi_heads
(cfg, dim_in)
single_modality/action_detection/alphaction/modeling/roi_heads/roi_heads_3d.py:27
↓ 1 callersFunctionbuild_backbone
(cfg)
single_modality/action_detection/alphaction/modeling/backbone/backbone.py:18
↓ 1 callersFunctionbuild_bert
build text encoder. Args: model_config (dict): model config. pretrain (bool): Whether to do pretrain or finetuning. check
multi_modality/models/backbones/bert/builder.py:6
↓ 1 callersFunctionbuild_bert_decoder
build text decoder the same as the multimodal encoder. Args: model_config (dict): model config. pretrain (bool): Whether to do pr
multi_modality/models/backbones/bert/builder.py:42
↓ 1 callersMethodbuild_data_multi_img_gt
each text may have multiple ground_truth image, e.g., ssv2
multi_modality/dataset/caption_dataset.py:108
↓ 1 callersMethodbuild_data_multi_txt_gt
each image may have multiple ground_truth text, e.g., COCO and Flickr30K
multi_modality/dataset/caption_dataset.py:122
↓ 1 callersFunctionbuild_dataset
Arguments: cfg: config object for the experiment. dataset_list (list[str]): Contains the names of the datasets, i.e.,
single_modality/action_detection/alphaction/dataset/build.py:15
↓ 1 callersFunctionbuild_object_transforms
(cfg, is_train=True)
single_modality/action_detection/alphaction/dataset/transforms/build.py:53
↓ 1 callersFunctionbuild_roi_action_head
(cfg, dim_in)
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/action_head.py:93
↓ 1 callersMethodbuild_text_decoder
(self)
multi_modality/models/umt_qa.py:42
↓ 1 callersMethodbuild_text_encoder
build text_encoder and possiblly video-to-text multimodal fusion encoder. Returns: nn.Module. The text encoder
multi_modality/models/umt.py:262
↓ 1 callersFunctionbuild_transforms
(cfg, is_train=True)
single_modality/action_detection/alphaction/dataset/transforms/build.py:5
↓ 1 callersMethodbuild_vision_encoder
build vision encoder Returns: (vision_encoder, clip_teacher). Each is a `nn.Module`.
multi_modality/models/umt.py:231
↓ 1 callersMethodclear
Clears the state to prepare for a fresh evaluation.
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/object_detection_evaluation.py:97
↓ 1 callersMethodclip_contrastive_temperature
Seems only used during pre-training
multi_modality/models/umt.py:227
↓ 1 callersMethodcompute_object_detection_metrics
Evaluates detections as being tp, fp or ignored from a single image. The evaluation is done in two stages: 1. All detections are matched to
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/per_image_evaluation.py:50
↓ 1 callersFunctioncompute_on_dataset
(model, data_loader, device, logger, mem_active)
single_modality/action_detection/alphaction/engine/inference.py:124
↓ 1 callersFunctioncompute_on_dataset_1stage
(model, data_loader, device)
single_modality/action_detection/alphaction/engine/inference.py:15
↓ 1 callersFunctioncompute_on_dataset_2stage
(model, data_loader, device, logger)
single_modality/action_detection/alphaction/engine/inference.py:41
↓ 1 callersFunctionconcatenate
Concatenate list of BoxLists. This op concatenates a list of input BoxLists into a larger BoxList. It also handles concatenation of BoxList fiel
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list_ops.py:423
↓ 1 callersFunctioncontrast_jitter
Perfrom contrast jittering on the input images. The channels of images should be in order BGR. Args: var (float): jitter ratio fo
single_modality/datasets/video_transforms.py:367
↓ 1 callersFunctionconvert_label_map_to_categories
Loads label map proto and returns categories list compatible with eval. This function loads a label map and returns a list of dicts, each of which
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/label_map_util.py:69
↓ 1 callersMethodconvert_to_roi_format
(self, boxes, dtype, device)
single_modality/action_detection/alphaction/modeling/poolers.py:21
↓ 1 callersFunctioncreate_category_index
Creates dictionary of COCO compatible categories keyed by category id. Args: categories: a list of dicts, each of which has the following keys:
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/label_map_util.py:38
↓ 1 callersFunctioncreate_optimizer
(args, model, filter_bias_and_bn=True)
multi_modality/utils/optimizer.py:87
↓ 1 callersFunctioncreate_optimizer
(args, model, get_num_layer=None, get_layer_scale=None, filter_bias_and_bn=True, skip_list=None,
single_modality/action_detection/optim_factory.py:103
↓ 1 callersFunctioncreate_optimizer_params_group
named_param_tuples_with_lr: List([name, param, weight_decay, lr])
multi_modality/utils/optimizer.py:65
↓ 1 callersFunctioncreate_scheduler
(args, optimizer)
multi_modality/utils/scheduler.py:9
↓ 1 callersFunctiondecode_image_key
(image_key)
single_modality/action_detection/data/ava_eval.py:41
↓ 1 callersFunctiondecode_image_key
(image_key)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/ava_eval.py:38
↓ 1 callersMethoddelete_field
(self, field)
single_modality/action_detection/alphaction/structures/bounding_box.py:48
↓ 1 callersFunctiondo_ava_evaluation
(dataset, predictions, output_folder)
single_modality/action_detection/data/ava_eval.py:14
↓ 1 callersFunctiondo_ava_evaluation
(dataset, predictions, output_folder, logger)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/ava_eval.py:11
↓ 1 callersMethodencode_teacher
encode image / videos as features. Args: image (torch.Tensor): The input images. Returns: tuple. - mask (tor
multi_modality/models/umt.py:117
↓ 1 callersMethodencode_vision
encode image / videos as features. Args: image (torch.Tensor): The input images. test (bool): Whether testing.
multi_modality/models/umt_qa.py:53
↓ 1 callersFunctioneval_after_training
(train_config)
multi_modality/tasks/retrieval.py:245
↓ 1 callersFunctioneval_after_training
(train_config)
multi_modality/tasks/vqa.py:334
↓ 1 callersFunctioneval_dict_leaf
eval values of dict leaf. Args: d (dict): The dict to eval. Returns: dict.
multi_modality/utils/config.py:222
↓ 1 callersFunctioneval_qa_acc
(gt_data_or_path, pred_data_or_path, is_vqa=False)
multi_modality/tasks/vqa_utils.py:282
↓ 1 callersFunctioneval_simple_qa_acc
For open-ended QA that has a single answer, accuracy is computed based on exact match
multi_modality/tasks/vqa_utils.py:260
↓ 1 callersFunctioneval_string
automatically evaluate string to corresponding types. For example: not a string -> return the original input '0' -> 0 '
multi_modality/utils/config.py:241
↓ 1 callersFunctionevaluate
evaluate dataset using different methods based on dataset type. Args: dataset: Dataset object predictions(list[BoxList]): each ite
single_modality/action_detection/alphaction/dataset/datasets/evaluation/__init__.py:6
↓ 1 callersMethodevaluate
(self, pred_data_or_path)
multi_modality/tasks/vqa_utils.py:136
↓ 1 callersFunctionevaluate_predictions_on_ava
(eval_file_paths, ava_results, csv_result_file)
single_modality/action_detection/data/ava_eval.py:194
↓ 1 callersFunctionevaluate_predictions_on_ava
(eval_file_paths, ava_results, csv_result_file, logger)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/ava_eval.py:191
↓ 1 callersFunctionevaluation
(model, data_loader, tokenizer, device, config)
multi_modality/tasks/retrieval_utils.py:92
↓ 1 callersFunctionevaluation
(model, data_loader, tokenizer, device, config)
multi_modality/tasks/vqa.py:106
↓ 1 callersFunctionextract_text_feats
(texts, max_txt_l, tokenizer, model, device)
multi_modality/tasks/retrieval_utils.py:18
↓ 1 callersFunctionextract_vision_feats
(data_loader, model, device, config)
multi_modality/tasks/retrieval_utils.py:43
↓ 1 callersMethodfill_fix_offset
(more_fix_crop, image_w, image_h, crop_w, crop_h)
single_modality/datasets/transforms.py:165
↓ 1 callersMethodflush
(self)
single_modality/action_detection/utils.py:187
↓ 1 callersMethodforward_features
(self, x, mask=None, use_image=False)
multi_modality/models/backbones/vit/vit.py:246
↓ 1 callersMethodforward_features
(self, x, proposals)
single_modality/action_detection/modeling_finetune.py:282
↓ 1 callersMethodforward_features
(self, x)
single_modality/models/modeling_finetune.py:303
↓ 1 callersMethodforward_features
(self, x, mask)
single_modality/models/modeling_pretrain.py:80
↓ 1 callersMethodforward_features
(self, x, mask)
single_modality/models/modeling_pretrain_umt.py:104
↓ 1 callersMethodfrom_file
Build config from file. Supported filetypes: `.py`,`.yaml`,`.json`. Args: filepath (str): The config file path. Returns:
multi_modality/utils/config.py:110
↓ 1 callersMethodfusion
(self, x, out, type="add")
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/roi_action_feature_extractor.py:176
↓ 1 callersFunctiongather
Run gather on arbitrary picklable data (not necessarily tensors). Args: data: any picklable object dst (int): destination ran
single_modality/action_detection/alphaction/utils/comm.py:133
↓ 1 callersFunctiongeometric_progression
(a, r, n)
multi_modality/models/utils.py:140
↓ 1 callersMethodget_anno
obtain the annotation for one media (video or image) Args: index (int): The media index. Returns: dict. - "i
multi_modality/dataset/base_dataset.py:42
↓ 1 callersMethodget_anno
(self, index)
multi_modality/dataset/sqlite_dataset.py:59
↓ 1 callersMethodget_answers_with_weights
(self, raw_answers)
multi_modality/dataset/qa_dataset.py:28
↓ 1 callersFunctionget_args
()
single_modality/run_umt_pretraining.py:22
↓ 1 callersFunctionget_args
()
single_modality/run_mae_pretraining.py:22
↓ 1 callersFunctionget_args
()
single_modality/run_class_finetuning.py:27
↓ 1 callersFunctionget_args
()
single_modality/action_detection/run_class_finetuning.py:23
↓ 1 callersMethodget_checkpoint_file
(self)
single_modality/action_detection/alphaction/utils/checkpoint.py:89
↓ 1 callersMethodget_config
get a `Config` instance. Args: default_config (dict): The default config. `default_config` will be overrided by c
multi_modality/utils/config.py:70
↓ 1 callersMethodget_coordinates
Get corner coordinates of boxes. Returns: a list of 4 1-d numpy arrays [y_min, x_min, y_max, x_max]
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:106
↓ 1 callersFunctionget_cosine_schedule_with_warmup
Modified from https://github.com/huggingface/transformers/blob/v4.15.0/src/transformers/optimization.py Create a schedule with a learning ra
multi_modality/utils/scheduler.py:22
↓ 1 callersMethodget_extended_attention_mask
Makes broadcastable attention and causal masks so that future and masked tokens are ignored. Arguments: attention_mask (
multi_modality/models/backbones/bert/xbert.py:1049
↓ 1 callersFunctionget_fast_model_cfg
(cfg)
single_modality/action_detection/alphaction/modeling/backbone/slowfast.py:51
↓ 1 callersMethodget_gather_args
obtain the args for all_gather Returns: dict.
multi_modality/models/criterions.py:197
↓ 1 callersFunctionget_grad_norm_
(parameters, norm_type: float = 2.0)
single_modality/utils.py:432
↓ 1 callersFunctionget_grad_norm_
(parameters, norm_type: float = 2.0)
single_modality/action_detection/utils.py:370
↓ 1 callersMethodget_idxs
(self, idx)
single_modality/action_detection/alphaction/dataset/datasets/concat_dataset.py:12
↓ 1 callersFunctionget_loss_scale_for_deepspeed
(model)
single_modality/engines/engine_for_finetuning.py:20
↓ 1 callersMethodget_memory_feature
(self, feature_pool, extras, mem_len, mem_rate, max_boxes, fixed_dim, current_x, current_box, use_penalty)
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/roi_action_feature_extractor.py:117
↓ 1 callersFunctionget_model
(args)
single_modality/run_umt_pretraining.py:159
↓ 1 callersFunctionget_model
(args)
single_modality/run_mae_pretraining.py:134
↓ 1 callersFunctionget_model_cfg
(cfg)
single_modality/action_detection/alphaction/modeling/backbone/i3d.py:9
↓ 1 callersFunctionget_num_layer_for_vit
(var_name, num_max_layer)
single_modality/optim_factory.py:24
↓ 1 callersFunctionget_num_layer_for_vit
(var_name, num_max_layer)
single_modality/action_detection/optim_factory.py:24
↓ 1 callersMethodget_num_layers
(self)
single_modality/action_detection/modeling_finetune.py:268
↓ 1 callersMethodget_num_layers
(self)
single_modality/models/modeling_finetune.py:289
↓ 1 callersMethodget_objects
(self, idx, im_w, im_h)
single_modality/action_detection/alphaction/dataset/datasets/ava.py:201
↓ 1 callersMethodget_params
Get parameters for ``crop`` for a random sized crop. Args: img (PIL Image): Image to be cropped. scale (tuple): range
single_modality/datasets/video_transforms.py:726
↓ 1 callersMethodget_params
(self, brightness, contrast, saturation, hue)
single_modality/datasets/video_transforms.py:1190
↓ 1 callersFunctionget_position_angle_vec
(position)
single_modality/action_detection/modeling_finetune.py:164
↓ 1 callersFunctionget_position_angle_vec
(position)
single_modality/models/modeling_finetune.py:161
← previousnext →301–400 of 1,198, ranked by callers