MCPcopy Create free account

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

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

↓ 29 callersMethodget
Returns the instance specified by the name. Args: name (str): name of the specified instance.
tadaconv/utils/registry.py:51
↓ 12 callersMethodseconds
Returns: (float): the total number of seconds since the start/reset of the timer, excluding the time when the
tadaconv/utils/timer.py:52
↓ 12 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
tadaconv/models/utils/init_helper.py:57
↓ 11 callersMethoditer_toc
Stop to record time.
tadaconv/utils/meters.py:725
↓ 10 callersMethoditer_tic
Start to record time.
tadaconv/utils/meters.py:719
↓ 10 callersMethodupdate_stats
Update the current stats. Args: top1_err (float): top1 error rate. top5_err (float): top5 error rate.
tadaconv/utils/meters.py:731
↓ 9 callersMethod__init__
Args: cfg (Config): global config object.
tadaconv/models/base/base_blocks.py:372
↓ 9 callersMethodadd_value
Add a new scalar value to the deque.
tadaconv/utils/meters.py:494
↓ 8 callersMethod_get_object_to_file
Wrapper for downloading the file object. Args: obj_file (str): the target file to be downloaded (if it starts
tadaconv/datasets/base/base_dataset.py:416
↓ 8 callersMethodget_win_median
Calculate the current median value of the deque.
tadaconv/utils/meters.py:502
↓ 8 callersMethodlog_iter_stats
log the stats of the current iteration. Args: cur_epoch (int): the number of current epoch. cur_iter (int): t
tadaconv/utils/meters.py:772
↓ 7 callersFunction_check_args_tf
(kwargs)
tadaconv/datasets/utils/auto_augment.py:45
↓ 7 callersMethodreset
Reset the Meter.
tadaconv/utils/meters.py:705
↓ 7 callersMethodupdate_custom_stats
Update stats using custom keys. Args: stats (dict): additional stats to be updated.
tadaconv/utils/meters.py:749
↓ 6 callersFunctioncutmix_bbox_and_lam
Generate bbox and apply lambda correction.
tadaconv/datasets/utils/mixup.py:90
↓ 6 callersMethoddump
(self)
tadaconv/utils/config.py:261
↓ 6 callersFunctionload_checkpoint
Load the checkpoint from the given file. Args: path_to_checkpoint (string): path to the checkpoint to load. model (model): m
tadaconv/utils/checkpoint.py:276
↓ 6 callersMethodreset
Reset the timer.
tadaconv/utils/timer.py:18
↓ 5 callersMethod__init__
( self, dim, num_heads=12, attn_dropout=0., ff_dropout=0., ein
tadaconv/models/base/transformer.py:52
↓ 5 callersMethod_construct_slowfast_module
(self, cfgs, module, **kwargs)
tadaconv/models/base/slowfast.py:52
↓ 5 callersFunction_get_oss_path_prefix
Get full path with epoch and subset Args: prefix (str): path prefix. epoch (int): epoch number of these proposals tes
tadaconv/utils/tal_tools.py:20
↓ 5 callersFunction_randomly_negate
With 50% prob, negate the value
tadaconv/datasets/utils/auto_augment.py:168
↓ 5 callersFunctionbuild_loader
Constructs the data loader for the given dataset. Args: cfg (Configs): global config object. details in utils/config.py split
tadaconv/datasets/base/builder.py:47
↓ 5 callersMethoddeep_copy
(self)
tadaconv/utils/config.py:264
↓ 5 callersFunctionget_checkpoint_dir
Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.
tadaconv/utils/checkpoint.py:79
↓ 5 callersMethodset_spatial_index
(self, idx)
tadaconv/datasets/utils/transformations.py:468
↓ 4 callersMethod__init__
Args: cfg (Config): global config object.
tadaconv/models/base/backbone.py:36
↓ 4 callersMethod_check_input
(self, value, name, center=1, bound=(0, float('inf')), clip_first_on_zero=True)
tadaconv/datasets/utils/transformations.py:52
↓ 4 callersMethod_get_sample_info
Returns the sample info corresponding to the index. Args: index (int): target index Returns: sample_
tadaconv/datasets/base/ssv2.py:40
↓ 4 callersFunction_init_convnet_weights
Performs ResNet style weight initialization. Args: fc_init_std (float): the expected standard deviation for fc layer. zero_in
tadaconv/models/utils/init_helper.py:125
↓ 4 callersMethod_interval_based_sampling
Generates the frame index list using interval based sampling. Args: vid_length (int): the length of the whole video (
tadaconv/datasets/base/base_dataset.py:486
↓ 4 callersFunction_is_tensor_a_torch_image
(input)
tadaconv/datasets/utils/transformations.py:149
↓ 4 callersMethod_merge_cfg_from_base
Replace the attributes in the base config by the values in the coming config, unless preserve base is set to True. Args:
tadaconv/utils/config.py:154
↓ 4 callersMethod_params_per_elem
(self, batch_size)
tadaconv/datasets/utils/mixup.py:137
↓ 4 callersFunctionbuild_model
Builds the video model. Args: cfg (Config): global config object that provides specifics to construct the model. gpu_id (Opti
tadaconv/models/base/builder.py:19
↓ 4 callersFunctionget_loss_func
Retrieve the loss given the loss name. Args (int): loss_name: the name of the loss to use.
tadaconv/models/utils/losses.py:41
↓ 4 callersMethodpause
Pause the timer.
tadaconv/utils/timer.py:27
↓ 4 callersMethodset_model_ema_enabled
(self, model_ema_enabled)
tadaconv/utils/meters.py:175
↓ 4 callersFunctionstr2sec
Transfer epic annotation strings to time. Args: instr (string): Time string read from annotation files.
tadaconv/datasets/base/epickitchen100_feature.py:41
↓ 3 callersFunction_blend
(img1, img2, ratio)
tadaconv/datasets/utils/transformations.py:152
↓ 3 callersMethod_get_predictions_with_label
Get all predicitons of the given label. Return empty DataFrame if there is no predcitions with the given label.
tadaconv/utils/eval_tal/eval_epic_detection.py:178
↓ 3 callersMethod_path_join
Join a list of paths. Args: path_list (list): list of paths.
tadaconv/utils/config.py:69
↓ 3 callersMethod_transform_feature_scale
We divide each videos to multi clips, and each clip contains 5s. For example, if the seg_time is [8.5, 16.1], which means that we nee
tadaconv/datasets/base/epickitchen100_feature.py:381
↓ 3 callersMethodget_interp1d_mask
generate sample mask for each point in Boundary-Matching Map Args: prop_boundary_ratio (float): Boundary expand ratio.
tadaconv/models/module_zoo/heads/bmn_head.py:161
↓ 3 callersFunctionioa_with_anchors
calculate the overlap proportion between the anchor and all bbox for supervise signal, Args: anchors_min (np.ndarry): 1d anchors star
tadaconv/utils/bboxes_1d.py:4
↓ 3 callersFunctionone_hot
(x, num_classes, on_value=1., off_value=0., device='cuda')
tadaconv/datasets/utils/mixup.py:13
↓ 3 callersMethodprint_map
(self, ap, _type)
tadaconv/utils/eval_tal/eval_epic_detection.py:244
↓ 3 callersFunctionrgb_to_grayscale
Convert the given RGB Image Tensor to Grayscale. For RGB to Grayscale conversion, ITU-R 601-2 luma transform is performed which is L = R * 0
tadaconv/datasets/utils/transformations.py:157
↓ 3 callersMethodupdate
(self, model)
tadaconv/models/utils/model_ema.py:44
↓ 3 callersMethodwrapper_compute_average_precision
Computes average precision for each class in the subset.
tadaconv/utils/eval_tal/eval_epic_detection.py:188
↓ 2 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
tadaconv/models/module_zoo/ops/tadaconv.py:69
↓ 2 callersMethod__init__
(self, cfg)
tadaconv/models/module_zoo/branches/tadaformer.py:62
↓ 2 callersMethod__init__
(self,cfg,in_planes,out_planes,kernel_size,stride,padding=0)
tadaconv/models/module_zoo/branches/s3dg_branch.py:94
↓ 2 callersMethod__init__
Args: cfg (Config): the global config object.
tadaconv/models/module_zoo/heads/mosi_head.py:18
↓ 2 callersMethod_construct_branch
( self, cfg, in_planes, out_planes, kernel_size, stride,
tadaconv/models/module_zoo/branches/s3dg_branch.py:114
↓ 2 callersMethod_erase
(self, img, chan, num_frames, img_h, img_w, dtype)
tadaconv/datasets/utils/random_erasing.py:71
↓ 2 callersMethod_mix_batch_x2
(self, x1, x2)
tadaconv/datasets/utils/mixup.py:270
↓ 2 callersMethod_mix_elem_x2
(self, x1, x2)
tadaconv/datasets/utils/mixup.py:225
↓ 2 callersMethod_mix_pair_x2
(self, x1, x2)
tadaconv/datasets/utils/mixup.py:245
↓ 2 callersMethod_params_per_batch
(self)
tadaconv/datasets/utils/mixup.py:157
↓ 2 callersMethod_read_feature
Load feature from file. Args: feature_path (str): File path for a feature. num_retries (int): Retry time if d
tadaconv/datasets/base/epickitchen100_feature.py:425
↓ 2 callersMethod_update
(self, model, update_fn)
tadaconv/models/utils/model_ema.py:37
↓ 2 callersFunctionbi_loss
(pred_score, gt_label, label_weight)
tadaconv/models/utils/localization_losses.py:30
↓ 2 callersFunctioncheckpoint_preprocess
Preprocess the checkpoints for video vision transformers. Enabled in the cfg. It includes processing the positional embedding by repeating an
tadaconv/utils/checkpoint.py:183
↓ 2 callersFunctiondrop_path
From https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/layers/drop.py. Drop paths (Stochastic Depth) per sample (when
tadaconv/models/base/base_blocks.py:509
↓ 2 callersFunctioneval_epoch
Evaluate the model on the val set. Args: val_loader (loader): data loader to provide validation data. model (model): model/mo
runs/train.py:197
↓ 2 callersMethodfinalize_metrics
Calculate and log the final ensembled metrics. ks (tuple): list of top-k values for topk_accuracies. For example, ks = (1
tadaconv/utils/meters.py:140
↓ 2 callersFunctionfuse_epic_sliding_windows
Since different sliding windows have overlap, we desigin this function to fuse sliding windows. Args: maps_dict (dict): {
runs/test_epic_localization.py:29
↓ 2 callersMethodget_crop_params
Returns crop parameters. Args: speed_factor (float): frac{distance_to_go}{total_distance} total_length (i
tadaconv/sslgenerators/mosi/mosi_generator.py:224
↓ 2 callersMethodget_global_avg
Calculate the global mean value.
tadaconv/utils/meters.py:514
↓ 2 callersFunctionget_last_checkpoint
Get the last checkpoint from the checkpointing folder. Args: path_to_job (string): the path to the folder of the current job.
tadaconv/utils/checkpoint.py:99
↓ 2 callersFunctionget_lr_func
Given the configs, retrieve the specified lr policy function. Args: lr_policy (string): the learning rate policy to use for the job.
tadaconv/models/utils/lr_policy.py:97
↓ 2 callersFunctionget_model_stats
Compute statistics for the current model given the config. Args: model (model): model to perform analysis. cfg (Config): the
tadaconv/utils/misc.py:103
↓ 2 callersFunctionhas_checkpoint
Determines if the given directory contains a checkpoint. Args: path_to_job (string): the path to the folder of the current job.
tadaconv/utils/checkpoint.py:115
↓ 2 callersFunctioniou_with_anchors
Compute jaccard score between a box and the anchors. Args: anchors_min (np.ndarry): 1d anchors start position, shape is N. an
tadaconv/utils/bboxes_1d.py:23
↓ 2 callersFunctionlabel_smoothing_target
(target, num_classes, smoothing=0.0, device='cuda')
tadaconv/datasets/utils/mixup.py:25
↓ 2 callersMethodlog_epoch_stats
Log the stats of the current epoch. Args: cur_epoch (int): the number of current epoch.
tadaconv/utils/meters.py:797
↓ 2 callersFunctionmixup_target
(target, num_classes, lam=1., smoothing=0.0, device='cuda')
tadaconv/datasets/utils/mixup.py:18
↓ 2 callersFunctionnorm_cdf
(x)
tadaconv/models/utils/init_helper.py:24
↓ 2 callersFunctionperform_test
Perform multi-view test on the specified test set, where {cfg.TEST.NUM_ENSEMBLE_VIEWS} clips and {cfg.TEST.NUM_SPATIAL_CROPS} crops are sampl
runs/test.py:25
↓ 2 callersFunctionperform_test
Perform sliding windows test on the specified test set Args: test_loader (loader): video testing loader. model (model): the p
runs/test_epic_localization.py:70
↓ 2 callersMethodset_model_ema_enabled
(self, model_ema_enabled)
tadaconv/utils/meters.py:827
↓ 2 callersFunctiontopks_correct
Given the predictions, labels, and a list of top-k values, compute the number of correct predictions for each top-k value. Args:
tadaconv/utils/metrics.py:100
↓ 2 callersMethodtrain
r"""Sets the module in training mode. This has any effect only on certain modules. See documentations of particular modules for detai
tadaconv/models/base/models.py:47
↓ 2 callersFunctiontransform
(x, y, matrix)
tadaconv/datasets/utils/auto_augment.py:101
↓ 2 callersFunctionupdate_3d_conv_params
Automatically decodes parameters for 3D convolution blocks according to the config and its index in the model. Args: cfg (Config):
tadaconv/models/utils/params.py:6
↓ 2 callersMethodupdate_predictions
Update predictions and labels. Args: preds (tensor): model output predictions. labels (tensor): labels.
tadaconv/utils/meters.py:761
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
tadaconv/models/module_zoo/ops/tadaconv_v2.py:125
↓ 1 callersMethod__init__
Args: cfg (Config): global config object.
tadaconv/models/module_zoo/stems/embedding_stem.py:21
↓ 1 callersMethod__init__
Args: cfg (Config): global config object.
tadaconv/models/module_zoo/heads/transformer_head.py:20
↓ 1 callersMethod__init__
Args: cfg (Config): global config object.
tadaconv/models/module_zoo/heads/slowfast_head.py:18
↓ 1 callersMethod__init__
(self, cfg)
tadaconv/models/base/slowfast.py:23
↓ 1 callersMethod__init__
(self, cfg)
tadaconv/models/base/models.py:71
↓ 1 callersMethod__init_dist__
(self)
tadaconv/utils/val_dist_sampler.py:79
↓ 1 callersMethod_apply_basic
(self, img, mixing_weights, m)
tadaconv/datasets/utils/auto_augment.py:790
↓ 1 callersMethod_apply_blended
(self, img, mixing_weights, m)
tadaconv/datasets/utils/auto_augment.py:774
↓ 1 callersMethod_boundary_matching_layer
Apply boundary mathcing operation for input feature Args: x (tensor): 1D feature for boundary mathcing operation.
tadaconv/models/module_zoo/heads/bmn_head.py:149
↓ 1 callersMethod_calc_blended_weights
(self, ws, m)
tadaconv/datasets/utils/auto_augment.py:764
↓ 1 callersMethod_config_transform
Configs the transform for the dataset. For train, we apply random cropping, random horizontal flip, random color jitter (optionally
tadaconv/datasets/base/base_dataset.py:551
↓ 1 callersMethod_construct_backbone
(self, cfg)
tadaconv/models/base/slowfast.py:66
next →1–100 of 571, ranked by callers