MCPcopy Create free account

hub / github.com/alibaba-mmai-research/TAdaConv / functions

Functions571 in github.com/alibaba-mmai-research/TAdaConv

↓ 1 callersMethod_construct_backbone
(self, cfg)
tadaconv/models/base/backbone.py:44
↓ 1 callersMethod_construct_backbone
( self, cfg, input_channel )
tadaconv/models/base/backbone.py:110
↓ 1 callersMethod_construct_backbone
( self, cfg, input_channel )
tadaconv/models/base/backbone.py:198
↓ 1 callersMethod_construct_block
( self, cfg, block_idx, )
tadaconv/models/base/base_blocks.py:128
↓ 1 callersMethod_construct_block
( self, cfg, dim_in, num_filters, kernel_sz, stride,
tadaconv/models/base/base_blocks.py:274
↓ 1 callersMethod_construct_block
( self, cfg, dim_in, num_filters, kernel_sz, stride,
tadaconv/models/base/base_blocks.py:340
↓ 1 callersMethod_construct_bottleneck
(self)
tadaconv/models/base/base_blocks.py:97
↓ 1 callersMethod_construct_branch
(self)
tadaconv/models/base/base_blocks.py:84
↓ 1 callersMethod_construct_dataset
Constructs the dataset according to the global config object. Currently supports reading from csv, json and txt. Args:
tadaconv/datasets/base/base_dataset.py:160
↓ 1 callersMethod_construct_head
( self, dim, num_classes, dropout_rate, activation_func )
tadaconv/models/module_zoo/heads/slowfast_head.py:51
↓ 1 callersMethod_construct_head
( self, dim, num_classes, dropout_rate, activation_func )
tadaconv/models/module_zoo/heads/slowfast_head.py:144
↓ 1 callersMethod_construct_head
( self, )
tadaconv/models/module_zoo/heads/bmn_head.py:44
↓ 1 callersMethod_construct_head
( self, dim, num_classes, dropout_rate, activation_func )
tadaconv/models/base/base_blocks.py:393
↓ 1 callersMethod_construct_simple_block
(self)
tadaconv/models/base/base_blocks.py:93
↓ 1 callersMethod_construct_slowfast_cfg
(self, cfg)
tadaconv/models/base/slowfast.py:37
↓ 1 callersMethod_construct_stage
( self, cfg, stage_idx, )
tadaconv/models/base/base_blocks.py:181
↓ 1 callersMethod_construct_video_anno
Transfer string to time or label.
tadaconv/datasets/base/epickitchen100_feature.py:181
↓ 1 callersMethod_construct_video_clips
We divide each videos to multi-clips, and each clip contains 5 seconds. This function is used to generate video clips path for easy
tadaconv/datasets/base/epickitchen100_feature.py:193
↓ 1 callersMethod_custom_sampling
(self, vid_length, vid_fps, clip_idx, num_clips, num_frames, interval=2, random_sample=True)
tadaconv/datasets/base/base_dataset.py:134
↓ 1 callersMethod_download_anno_json
Init annotation files and video-duration list for evaluation.
tadaconv/datasets/base/epickitchen100_feature.py:81
↓ 1 callersFunction_download_clip
from https://github.com/openai/CLIP/blob/main/clip/clip.py
tadaconv/utils/checkpoint.py:29
↓ 1 callersMethod_get_bmn_iou_map
Get the maps for boundary matching network. Args: gt_bbox (np.array): The shape is N*2. Normalized temporal bounding boxe
tadaconv/datasets/base/epickitchen100_feature.py:480
↓ 1 callersMethod_get_bmn_train_label
Get full bmn training labels. Args: gt_time (np.array): The shape is N*2. The ground truth temporal bounding boxes.
tadaconv/datasets/base/epickitchen100_feature.py:542
↓ 1 callersMethod_get_controlled_crop
(self, clip)
tadaconv/datasets/utils/transformations.py:405
↓ 1 callersMethod_get_dataset_list_name
Returns the list for the dataset. Returns: name (str): name of the list to be read
tadaconv/datasets/base/base_dataset.py:95
↓ 1 callersMethod_get_feature_files
We divide each videos to multi clips, and each clip contains 5s. This function is used to get clips which contain seg_time.
tadaconv/datasets/base/epickitchen100_feature.py:213
↓ 1 callersFunction_get_global_gloo_group
Return a process group based on gloo backend, containing all the ranks The result is cached. Returns: (group): pytorch dist group
tadaconv/utils/distributed.py:146
↓ 1 callersMethod_get_interp1d_bin_mask
generate sample mask for a boundary-matching pair Args: seg_xmin (float): Start time of the proposal. seg_xma
tadaconv/models/module_zoo/heads/bmn_head.py:192
↓ 1 callersFunction_get_model_analysis_input
Return a dummy input for model analysis with batch size 1. The input is used for analyzing the model (counting flops and activations etc.
tadaconv/utils/misc.py:69
↓ 1 callersFunction_get_pixels
(per_pixel, rand_color, patch_size, dtype=torch.float32, device='cuda')
tadaconv/datasets/utils/random_erasing.py:14
↓ 1 callersMethod_get_random_crop
(self, clip)
tadaconv/datasets/utils/transformations.py:447
↓ 1 callersMethod_get_sample_info
Get annotation and data path for training and validation. Args: index (int): the index of the data in the self._sa
tadaconv/datasets/base/epickitchen100_feature.py:243
↓ 1 callersMethod_get_sample_info
Returns the sample info corresponding to the index. Args: index (int): target index Returns: sa
tadaconv/datasets/base/base_dataset.py:104
↓ 1 callersMethod_get_start_end
Get the start and end sequences. Args: gt_bbox (np.array): The shape is N*2. Normalized temporal bounding boxes.
tadaconv/datasets/base/epickitchen100_feature.py:509
↓ 1 callersMethod_get_transform
Get a randomized transform to be applied on image. Arguments are same as that of __init__. Arg: T (int): number of fram
tadaconv/datasets/utils/transformations.py:71
↓ 1 callersMethod_get_video_frames_list
Returns the list of frame indexes in the video for decoding. Args: vid_length (int): video length clip_i
tadaconv/datasets/base/base_dataset.py:137
↓ 1 callersFunction_hsv2rgb
(img)
tadaconv/datasets/utils/transformations.py:202
↓ 1 callersMethod_import_ground_truth
Reads ground truth file, checks if it is well formatted, and returns the ground truth instances and the activity classes. Paramete
tadaconv/utils/eval_tal/eval_epic_detection.py:61
↓ 1 callersMethod_import_prediction
Reads prediction file, checks if it is well formatted, and returns the prediction instances. Parameters ----------
tadaconv/utils/eval_tal/eval_epic_detection.py:124
↓ 1 callersMethod_init_temporal_tools
Init temporal tools for boundary matching map generation. Such as anchors and valid iou mask for boundary matching network.
tadaconv/datasets/base/epickitchen100_feature.py:106
↓ 1 callersMethod_initialize_cfg
When loading config for the first time, base config is required to be read.
tadaconv/utils/config.py:80
↓ 1 callersMethod_initialize_oss
Initializes the oss bucket. Currently supporting two OSS accounts.
tadaconv/datasets/base/base_dataset.py:454
↓ 1 callersFunction_interpolation
(kwargs)
tadaconv/datasets/utils/auto_augment.py:37
↓ 1 callersMethod_load_oss_data
Load feature data and classification data. Args: sample_info (dict): feat_path for feature and cls_path for classificatio
tadaconv/datasets/base/epickitchen100_feature.py:334
↓ 1 callersMethod_load_video_length
Load the true length of videos for sliding windows. Args: cfg (Configs): global config object. details in utils/config.py
tadaconv/datasets/base/epickitchen100_feature.py:131
↓ 1 callersMethod_load_yaml
Load the specified yaml file. Args: args: parsed args by `self._parse_args`. file_name (str): the file name t
tadaconv/utils/config.py:95
↓ 1 callersMethod_merge_cfg_from_command
Merge cfg from command. Currently only support depth of four. E.g. VIDEO.BACKBONE.BRANCH.XXXX. is an attribute with depth of four.
tadaconv/utils/config.py:177
↓ 1 callersMethod_mix_batch
(self, x)
tadaconv/datasets/utils/mixup.py:212
↓ 1 callersMethod_mix_elem
(self, x)
tadaconv/datasets/utils/mixup.py:175
↓ 1 callersMethod_mix_pair
(self, x)
tadaconv/datasets/utils/mixup.py:192
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
tadaconv/models/utils/init_helper.py:21
↓ 1 callersMethod_norm_gt_boxes
Norm ground truth time into [0,1]. Args: gt_time (np.array): The shape is N*2. There have N ground truth action in seg_ti
tadaconv/datasets/base/epickitchen100_feature.py:460
↓ 1 callersFunction_pad_to_largest_tensor
Padding all the tensors from different GPUs to the largest ones. Args: tensor (tensor): tensor to pad. group (group): pytorch
tadaconv/utils/distributed.py:187
↓ 1 callersMethod_parse_args
Wrapper for argument parser.
tadaconv/utils/config.py:42
↓ 1 callersFunction_posterize_level_to_arg
(level, _hparams)
tadaconv/datasets/utils/auto_augment.py:215
↓ 1 callersMethod_pre_transformation_config
Set transformation parameters if required.
tadaconv/datasets/base/base_dataset.py:127
↓ 1 callersFunction_prepare_data
(cfg)
runs/run.py:20
↓ 1 callersMethod_process_localization_dataset
Generate sliding windows and matching their ground truth for training and validation. Args: cfg (Configs): global config
tadaconv/datasets/base/epickitchen100_feature.py:146
↓ 1 callersMethod_read_image
Wrapper for downloading the image and generating the PIL.Image object for reading the image. Args: path (str): im
tadaconv/datasets/base/base_dataset.py:276
↓ 1 callersMethod_read_video
Wrapper for downloading the video and generating the VideoReader object for reading the video. Args: video_path (str
tadaconv/datasets/base/base_dataset.py:204
↓ 1 callersMethod_register
Registers the instance. Args: name (str): name of the entry entry (): instance of the entry, could be any t
tadaconv/utils/registry.py:28
↓ 1 callersFunction_rgb2hsv
(img)
tadaconv/datasets/utils/transformations.py:181
↓ 1 callersMethod_segment_based_sampling
Generates the frame index list using segment based sampling. Args: vid_length (int): the length of the whole video
tadaconv/datasets/base/base_dataset.py:524
↓ 1 callersFunction_select_rand_weights
(weight_idx=0, transforms=None)
tadaconv/datasets/utils/auto_augment.py:617
↓ 1 callersFunction_serialize_to_tensor
Seriialize the tensor to ByteTensor. Note that only `gloo` and `nccl` backend is supported. Args: data (data): data to be ser
tadaconv/utils/distributed.py:159
↓ 1 callersFunction_solarize_level_to_arg
(level, _hparams)
tadaconv/datasets/utils/auto_augment.py:236
↓ 1 callersFunction_suppress_print
Suppresses printing from the current process.
tadaconv/utils/logging.py:21
↓ 1 callersMethod_update_dict
Set the dict to be attributes of the config recurrently. Args: cfg_dict (dict): the dictionary to be set as the attribute
tadaconv/utils/config.py:234
↓ 1 callersFunctionadjust_brightness
(img, brightness_factor)
tadaconv/datasets/utils/transformations.py:234
↓ 1 callersFunctionadjust_contrast
(img, contrast_factor)
tadaconv/datasets/utils/transformations.py:241
↓ 1 callersFunctionadjust_hue
Adjust hue of an image. The image hue is adjusted by converting the image to HSV and cyclically shifting the intensities in the hue channel
tadaconv/datasets/utils/transformations.py:257
↓ 1 callersFunctionadjust_saturation
(img, saturation_factor)
tadaconv/datasets/utils/transformations.py:250
↓ 1 callersMethodattention
(self, x: torch.Tensor)
tadaconv/models/module_zoo/branches/tadaformer.py:142
↓ 1 callersFunctionaugmix_ops
(magnitude=10, hparams=None, transforms=None)
tadaconv/datasets/utils/auto_augment.py:744
↓ 1 callersFunctionauto_augment_policy
(name='v0', hparams=None)
tadaconv/datasets/utils/auto_augment.py:492
↓ 1 callersFunctionauto_augment_policy_original
(hparams)
tadaconv/datasets/utils/auto_augment.py:426
↓ 1 callersFunctionauto_augment_policy_originalr
(hparams)
tadaconv/datasets/utils/auto_augment.py:459
↓ 1 callersFunctionauto_augment_policy_v0
(hparams)
tadaconv/datasets/utils/auto_augment.py:359
↓ 1 callersFunctionauto_augment_policy_v0r
(hparams)
tadaconv/datasets/utils/auto_augment.py:392
↓ 1 callersFunctionbuild_dataset
Builds a dataset according to the "dataset_name". Args: dataset_name (str): the name of the dataset to be constructed. cf
tadaconv/datasets/base/builder.py:113
↓ 1 callersFunctionbuild_ssl_generator
Entry point to registered self-supervised learning methods. Returns transformed frames and the self-supervised label. Args:
tadaconv/sslgenerators/builder.py:10
↓ 1 callersFunctionc2_msra_fill
Initialize `module.weight` using the "MSRAFill" implemented in Caffe2. Also initializes `module.bias` to 0. Args: module (torch.n
tadaconv/models/utils/init_helper.py:112
↓ 1 callersFunctioncompute_average_precision_detection
Compute average precision (detection task) between ground truth and predictions data frames. If multiple predictions occurs for the same predi
tadaconv/utils/eval_tal/eval_epic_detection.py:256
↓ 1 callersMethodconfig_transform
Configures the transformation applied to the pseudo motion sequence.
tadaconv/sslgenerators/mosi/mosi_generator.py:326
↓ 1 callersFunctionconvert_convnext_weights
(cfg, src, tgt)
tadaconv/utils/checkpoint.py:679
↓ 1 callersFunctionconvert_resnet_weights
(cfg, src, tgt)
tadaconv/utils/checkpoint.py:614
↓ 1 callersFunctionconvert_vit_clip_weights
(cfg, src, tgt)
tadaconv/utils/checkpoint.py:713
↓ 1 callersFunctioncreat_auto_augmentation
(aug_str, img_size, mean)
tadaconv/datasets/utils/auto_augment.py:714
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated mean average precision to measure the performance of a
tadaconv/utils/eval_tal/eval_epic_detection.py:231
↓ 1 callersFunctionevaluate_detection
Evaluate action detection performance. Args: video_anno (str): Annotation file path. detection_result_file (str): The detecti
tadaconv/utils/eval_tal/eval_tal.py:12
↓ 1 callersMethodforward_attention
(self, x)
tadaconv/models/module_zoo/ops/tadaconv_v2.py:104
↓ 1 callersMethodforward_features
(self, x)
tadaconv/models/base/backbone.py:370
↓ 1 callersMethodforward_partial
(self, x, einops_from, einops_to, **einops_dims)
tadaconv/models/base/transformer.py:101
↓ 1 callersMethodforward_wo_head
(self, x: torch.Tensor)
tadaconv/models/base/backbone.py:286
↓ 1 callersMethodframe_size_standardize
Standardize the frame size according to the settings in the cfg. Args: frame (Tensor): a single frame with the shape o
tadaconv/sslgenerators/mosi/mosi_generator.py:262
↓ 1 callersFunctionfuse_verb_noun_map
Simply fuse verb map and noun map for action classification Args: cfg (Config): the global config object. verb_map (dict): ve
tadaconv/utils/tal_tools.py:139
↓ 1 callersMethodgenerate_joint_labels
Generates labels for joint prediction.
tadaconv/sslgenerators/mosi/mosi_generator.py:208
↓ 1 callersMethodgenerate_separate_labels
Generates labels for separate prediction.
tadaconv/sslgenerators/mosi/mosi_generator.py:190
↓ 1 callersMethodget_args
Returns the read arguments.
tadaconv/utils/config.py:252
↓ 1 callersMethodget_crop
(self, clip)
tadaconv/datasets/utils/transformations.py:323
← previousnext →101–200 of 571, ranked by callers