MCPcopy Create free account

hub / github.com/OpenGVLab/unmasked_teacher / functions

Functions1,198 in github.com/OpenGVLab/unmasked_teacher

↓ 130 callersMethodto
(self, device)
single_modality/action_detection/alphaction/structures/bounding_box.py:258
↓ 112 callersFunctionprint
(*args, **kwargs)
single_modality/action_detection/utils.py:212
↓ 103 callersFunctionprint
(*args, **kwargs)
single_modality/utils.py:274
↓ 78 callersMethoditems
(self)
single_modality/action_detection/alphaction/structures/memory_pool.py:37
↓ 66 callersMethodupdate
(self, **kwargs)
single_modality/action_detection/utils.py:88
↓ 48 callersMethodget
(name)
single_modality/action_detection/alphaction/config/paths_catalog.py:30
↓ 39 callersMethodload
(self, f=None, model_weight_only=False, adjust_scheduler=False, no_head=False)
single_modality/action_detection/alphaction/utils/checkpoint.py:51
↓ 39 callersMethodtranspose
Transpose bounding box (flip or rotate in 90 degree steps) :param method: One of :py:attr:`PIL.Image.FLIP_LEFT_RIGHT`, :py:
single_modality/action_detection/alphaction/structures/bounding_box.py:132
↓ 35 callersFunctionis_main_process
()
multi_modality/utils/distributed.py:48
↓ 35 callersMethodstate_dict
(self)
single_modality/utils.py:425
↓ 29 callersMethodpop
(self, k, d=None)
multi_modality/utils/easydict.py:141
↓ 23 callersMethod__init__
(self, config, layer_num)
multi_modality/models/backbones/bert/xbert.py:600
↓ 21 callersMethodadd_field
(self, field, field_data)
single_modality/action_detection/alphaction/structures/bounding_box.py:39
↓ 21 callersMethodget_field
Accesses dataset associated with the specified field in the box collection. Args: field: a string parameter used to speficy a related field
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:90
↓ 21 callersMethodsave
(self, name, **kwargs)
single_modality/action_detection/alphaction/utils/checkpoint.py:31
↓ 19 callersMethodstate_dict
(self)
single_modality/action_detection/utils.py:363
↓ 18 callersMethodget
Convenience function for accesssing box coordinates. Returns: a numpy array of shape [N, 4] representing box corners
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:82
↓ 17 callersMethodupdate
(self, e=None, **f)
multi_modality/utils/easydict.py:135
↓ 15 callersMethodconvert
(self, mode)
single_modality/action_detection/alphaction/structures/bounding_box.py:58
↓ 15 callersFunctionget_rank
()
multi_modality/utils/distributed.py:42
↓ 13 callersMethodload_state_dict
(self, state_dict)
single_modality/action_detection/utils.py:366
↓ 12 callersMethodadd_field
Add dataset to a specified field. Args: field: a string parameter used to speficy a related field to be accessed. field_data: a numpy
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:65
↓ 12 callersMethodnum_boxes
Return number of boxes held in collections.
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:54
↓ 12 callersMethodupdate
(self, **kwargs)
multi_modality/utils/basic_utils.py:89
↓ 11 callersMethodadd_meter
(self, name, meter)
multi_modality/utils/basic_utils.py:139
↓ 11 callersMethodhas_field
(self, field)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:62
↓ 11 callersMethodmax
(self)
single_modality/action_detection/utils.py:67
↓ 10 callersMethodget_field
(self, field)
single_modality/action_detection/alphaction/structures/bounding_box.py:42
↓ 10 callersMethodload_state_dict
(self, state_dict)
single_modality/utils.py:428
↓ 9 callersFunction_cfg
(url='', **kwargs)
single_modality/models/modeling_finetune.py:11
↓ 9 callersMethodextend
Return a extended bounding box copy of this bounding box. All other fields should be keep unchanged. :param scale: By what ex
single_modality/action_detection/alphaction/structures/bounding_box.py:197
↓ 9 callersMethodget_masks
Convenience function for accessing masks. Returns: a numpy array of shape [N, height, width] representing masks
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_mask_list.py:56
↓ 9 callersFunctioninit_layer
(layer, init_std, bias)
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/IA_structure.py:396
↓ 9 callersFunctionlog_dict_to_wandb
include a separator `/` at the end of `prefix`
multi_modality/utils/logger.py:17
↓ 9 callersMethodlog_every
(self, iterable, log_freq, header=None)
multi_modality/utils/basic_utils.py:142
↓ 9 callersMethodresize
Returns a resized copy of this bounding box :param size: The requested size in pixels, as a 2-tuple: (width, height).
single_modality/action_detection/alphaction/structures/bounding_box.py:94
↓ 9 callersMethodset_epoch
(self, epoch)
single_modality/action_detection/alphaction/dataset/samplers/distributed.py:64
↓ 9 callersFunctiontrunc_normal_
(tensor, mean=0., std=1.)
single_modality/models/modeling_pretrain.py:13
↓ 8 callersMethodadd_meter
(self, name, meter)
single_modality/action_detection/utils.py:117
↓ 8 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
single_modality/action_detection/alphaction/modeling/utils.py:9
↓ 8 callersMethoddump
dump cfg to `json` file. Args: cfg (dict): The dict to dump. savepath (str): The filepath to save the dumped dict.
multi_modality/utils/config.py:55
↓ 8 callersFunctionget_world_size
()
multi_modality/utils/distributed.py:36
↓ 8 callersFunctionget_world_size
()
single_modality/action_detection/alphaction/utils/comm.py:12
↓ 8 callersMethodupdate
(self, **kwargs)
single_modality/action_detection/alphaction/utils/metric_logger.py:45
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
multi_modality/models/backbones/vit/vit.py:39
↓ 7 callersFunction_check_args_tf
(kwargs)
single_modality/datasets/rand_augment.py:58
↓ 7 callersFunctiongather
Gather boxes from BoxList according to indices and return new BoxList. By default, gather returns boxes corresponding to the input index list, as
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list_ops.py:94
↓ 7 callersMethodlog_every
(self, iterable, print_freq, header=None)
single_modality/action_detection/utils.py:120
↓ 7 callersFunctionsave_json
(data, filename, save_pretty=False, sort_keys=False)
multi_modality/utils/basic_utils.py:233
↓ 7 callersMethodtranspose_for_scores
(self, x)
multi_modality/models/backbones/bert/xbert.py:385
↓ 6 callersMethod__init__
Constructor. Args: categories: A list of dicts, each of which has the following keys - 'id': (required) an integer id uniquely iden
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/object_detection_evaluation.py:59
↓ 6 callersMethod_split_into_xyxy
(self)
single_modality/action_detection/alphaction/structures/bounding_box.py:78
↓ 6 callersMethodadd_scalar
(self, key, value, iteration)
multi_modality/utils/logger.py:244
↓ 6 callersFunctioncutmix_bbox_and_lam
Generate bbox and apply lambda correction.
single_modality/datasets/mixup.py:77
↓ 6 callersMethodload_frame
Load video content using Decord
single_modality/datasets/ssv2.py:252
↓ 6 callersMethodloadvideo_decord
Load video content using Decord
single_modality/datasets/kinetics.py:239
↓ 6 callersMethodloadvideo_decord
Load video content using Decord
single_modality/datasets/ssv2.py:532
↓ 6 callersMethodloadvideo_decord
Load video content using Decord
single_modality/datasets/kinetics_sparse.py:260
↓ 6 callersMethodmax
(self)
multi_modality/utils/basic_utils.py:68
↓ 6 callersFunctionpre_text
(text, max_l=None)
multi_modality/dataset/utils.py:68
↓ 6 callersFunctionrandom_short_side_scale_jitter
Perform a spatial short scale jittering on the given images and corresponding boxes. Args: images (tensor): images to perform sca
single_modality/datasets/video_transforms.py:43
↓ 6 callersMethodsynchronize_between_processes
(self)
single_modality/action_detection/utils.py:113
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
single_modality/action_detection/modeling_finetune.py:40
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
single_modality/models/modeling_finetune.py:36
↓ 5 callersFunction_randomly_negate
With 50% prob, negate the value
single_modality/datasets/rand_augment.py:195
↓ 5 callersFunctioncreate_dataset
(dataset_type, config)
multi_modality/dataset/__init__.py:24
↓ 5 callersMethodget_extra_fields
Return all non-box fields.
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_list.py:58
↓ 5 callersFunctionget_media_type
(dataset_config)
multi_modality/dataset/__init__.py:17
↓ 5 callersMethodload_and_transform_media_data
(self, index, data_path)
multi_modality/dataset/base_dataset.py:58
↓ 5 callersFunctionprepare_pooled_feature
(x_pooled, boxes, detach=True)
single_modality/action_detection/alphaction/modeling/utils.py:30
↓ 5 callersFunctionprint_time
(message, start)
single_modality/action_detection/data/ava_eval.py:190
↓ 5 callersFunctionprint_time
(logger, message, start)
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/ava_eval.py:187
↓ 5 callersMethodset_step
(self, step=None)
single_modality/action_detection/utils.py:172
↓ 5 callersFunctionsetup_seed
(seed)
multi_modality/utils/basic_utils.py:221
↓ 4 callersMethod__init__
(self, ann_file, transform, has_multi_vision_gt=False)
multi_modality/dataset/caption_dataset.py:85
↓ 4 callersMethod__init__
(self, dim_person, dim_mem, dim_out, structure_cfg)
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/IA_structure.py:132
↓ 4 callersFunction_get_param_spatial_crop
Given scale, ratio, height and width, return sampled coordinates of the videos.
single_modality/datasets/video_transforms.py:498
↓ 4 callersMethod_params_per_elem
(self, batch_size)
single_modality/datasets/mixup.py:121
↓ 4 callersMethodadd_single_ground_truth_image_info
Adds groundtruth for a single image to be used for evaluation. Args: image_id: A unique string/integer identifier for the image. grou
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/object_detection_evaluation.py:70
↓ 4 callersFunctionav_decode_video
(video_path)
single_modality/action_detection/alphaction/utils/video_decode.py:3
↓ 4 callersMethodbackward
(ctx, grad_output)
multi_modality/models/utils.py:219
↓ 4 callersMethodbackward
(ctx, grad_output)
single_modality/action_detection/alphaction/layers/roi_pool_3d.py:23
↓ 4 callersFunctionbox_list_to_box_mask_list
Converts a BoxList containing 'masks' into a BoxMaskList. Args: boxlist: An np_box_list.BoxList object. Returns: An np_box_mask_list.Box
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_mask_list_ops.py:27
↓ 4 callersMethodclip_to_image
(self, remove_empty=True)
single_modality/action_detection/alphaction/structures/bounding_box.py:300
↓ 4 callersFunctioncreate_loader
(datasets, samplers, batch_size, num_workers, is_trains, collate_fns)
multi_modality/dataset/__init__.py:238
↓ 4 callersFunctioncreate_random_augment
Get video randaug transform. Args: input_size: The size of the input video in tuple. auto_augment: Parameters for randaug. A
single_modality/datasets/video_transforms.py:620
↓ 4 callersMethodcrop
Cropss a rectangular region from this bounding box. The box is a 4-tuple defining the left, upper, right, and lower pixel coo
single_modality/action_detection/alphaction/structures/bounding_box.py:169
↓ 4 callersMethodencode_vision
encode image / videos as features. Args: image (torch.Tensor): The input images. test (bool): Whether testing.
multi_modality/models/umt.py:171
↓ 4 callersMethodflush
(self)
single_modality/utils.py:247
↓ 4 callersFunctiongather
Gather boxes from np_box_mask_list.BoxMaskList according to indices. By default, gather returns boxes corresponding to the input index list, as w
single_modality/action_detection/alphaction/dataset/datasets/evaluation/ava/pascal_evaluation/np_box_mask_list_ops.py:108
↓ 4 callersMethodget_global_avg_dict
include a separator (e.g., `/`, or "_") at the end of `prefix`
multi_modality/utils/basic_utils.py:130
↓ 4 callersFunctionget_sinusoid_encoding_table
Sinusoid position encoding table
multi_modality/models/backbones/vit/vit.py:163
↓ 4 callersMethodget_text_encoder
get text encoder, used for text and cross-modal encoding
multi_modality/models/umt.py:281
↓ 4 callersMethodglobal_avg
(self)
multi_modality/utils/basic_utils.py:117
↓ 4 callersFunctionhas_object
(ia_config)
single_modality/action_detection/alphaction/utils/IA_helper.py:13
↓ 4 callersFunctionis_dist_avail_and_initialized
()
multi_modality/utils/distributed.py:28
↓ 4 callersFunctionload_anno
[summary] Args: ann_file_list (List[List[str, str]] or List[str, str]): the latter will be automatically converted to the for
multi_modality/dataset/utils.py:31
↓ 4 callersFunctionload_specific_model
(model, model_ema, args, output_dir, model_name)
single_modality/utils.py:588
↓ 4 callersFunctionseparate_roi_per_person
:param things: [n2, c, t, h, w] :param proposals: :param max_things: :return [n, max_other, c, t, h, w]
single_modality/action_detection/alphaction/modeling/roi_heads/action_head/IA_structure.py:341
↓ 4 callersFunctionsetup_main
Setup config, logger, output_dir, etc. Shared for pretrain and all downstream tasks.
multi_modality/utils/config_utils.py:45
next →1–100 of 1,198, ranked by callers