MCPcopy Create free account

hub / github.com/OpenGVLab/UniFormerV2 / functions

Functions872 in github.com/OpenGVLab/UniFormerV2

Functionrandom_short_side_scale_jitter
Perform a spatial short scale jittering on the given images and corresponding boxes. Args: images (tensor): images to perform sca
slowfast/datasets/transform.py:41
Functionrandom_short_side_scale_jitter_list
Perform a spatial short scale jittering on the given images and corresponding boxes. Args: images (list): list of images to perfo
slowfast/datasets/cv2_transform.py:29
Functionrandom_sized_crop
Perform random sized cropping on the given image. Random crop with size 8% - 100% image area and aspect ratio in [3/4, 4/3]. Args:
slowfast/datasets/cv2_transform.py:513
Functionrandom_sized_crop_img
Performs Inception-style cropping (used for training).
slowfast/datasets/transform.py:693
Functionrandom_sized_crop_list
Perform random sized cropping on the given list of images. Random crop with size 8% - 100% image area and aspect ratio in [3/4, 4/3].
slowfast/datasets/cv2_transform.py:582
Methodreset
Reset the deque.
slowfast/utils/meters.py:423
Methodreset
Reset the Meter.
slowfast/utils/meters.py:487
Methodreset
Reset the Meter.
slowfast/utils/meters.py:633
Methodreset
Reset the epoch timer.
slowfast/utils/meters.py:791
Methodreset
Reset the deque.
slowfast/utils/meters_co.py:423
Methodreset
Reset the Meter.
slowfast/utils/meters_co.py:493
Methodreset
Reset the Meter.
slowfast/utils/meters_co.py:642
Methodreset
Reset the epoch timer.
slowfast/utils/meters_co.py:800
Methodreset_classifier
(self, num_classes, global_pool='')
slowfast/models/uniformer.py:381
Methodresult_available
How many results are ready to be returned.
slowfast/visualization/async_predictor.py:115
Methodresult_available
(self)
slowfast/visualization/async_predictor.py:225
Functionretry_load_images
This function is to load images with support of retrying for failed load. Args: image_paths (list): paths of images needed to be loa
slowfast/datasets/utils.py:19
Functionrevert_scaled_boxes
Revert scaled input boxes to match the original image size. Args: size (int): size of the cropped image. boxes (array): shape
slowfast/datasets/cv2_transform.py:783
Functionrevert_tensor_normalize
Revert normalization for a given tensor by multiplying by the std and adding the mean. Args: tensor (tensor): tensor to revert normal
slowfast/datasets/utils.py:383
Functionrgb2bgr
Convert clip frames from RGB mode to BRG mode. Args: x (Tensor): A tensor of the clip's RGB frames with shape: (channel,
slowfast/datasets/ptv_datasets.py:297
Functionrotate
(img, degrees, **kwargs)
slowfast/datasets/rand_augment.py:110
Functionrun
Runs a function from a child process. Args: local_rank (int): rank of the current process on the current machine. num_proc (i
slowfast/utils/multiprocessing.py:9
Methodrun
Run prediction asynchronously.
slowfast/visualization/async_predictor.py:41
Methodrun
Run visualization asynchronously.
slowfast/visualization/async_predictor.py:141
Methodsampler
Returns: (torch.utils.data.Sampler): video sampler for the dataset.
slowfast/datasets/ptv_datasets.py:68
Functionsave_best_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer_list (optim): optimizer to save
slowfast/utils/checkpoint_amp_co.py:161
Functionsave_best_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer (optim): optimizer to save the
slowfast/utils/checkpoint_amp_ema.py:159
Functionsave_best_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer (optim): optimizer to save the
slowfast/utils/checkpoint_amp.py:159
Functionsave_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer_list (optim): optimizer to save
slowfast/utils/checkpoint_amp_co.py:122
Functionsave_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer (optim): optimizer to save the
slowfast/utils/checkpoint.py:118
Functionsave_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer (optim): optimizer to save the
slowfast/utils/checkpoint_amp_ema.py:122
Functionsave_checkpoint
Save a checkpoint. Args: model (model): model to save the weight to the checkpoint. optimizer (optim): optimizer to save the
slowfast/utils/checkpoint_amp.py:122
Functionset_lr
Sets the optimizer lr to the specified value. Args: optimizer (optim): the optimizer using to optimize the current network. n
slowfast/models/optimizer.py:117
Functionset_lr
Sets the optimizer lr to the specified value. Args: optimizer (optim): the optimizer using to optimize the current network. n
slowfast/models/optimizer_backbone_special.py:155
Functionset_lr
Sets the optimizer lr to the specified value. Args: optimizer (optim): the optimizer using to optimize the current network. n
slowfast/models/optimizer_backbone.py:140
Functionsetup_logging
Sets up the logging for multiple processes. Only enable the logging for the master process, and suppress logging for the non-master processes
slowfast/utils/logging.py:40
Functionsharpness
(img, factor, **__)
slowfast/datasets/rand_augment.py:193
Functionshear_x
(img, factor, **kwargs)
slowfast/datasets/rand_augment.py:66
Functionshear_y
(img, factor, **kwargs)
slowfast/datasets/rand_augment.py:73
Functionshuffle_dataset
" Shuffles the data. Args: loader (loader): data loader to perform shuffle. cur_epoch (int): number of the current epoch.
slowfast/datasets/loader.py:173
Methodshutdown
(self)
slowfast/visualization/async_predictor.py:110
Methodshutdown
(self)
slowfast/visualization/async_predictor.py:220
Functionsolarize
(img, thresh, **__)
slowfast/datasets/rand_augment.py:156
Functionsolarize_add
(img, add, thresh=128, **__)
slowfast/datasets/rand_augment.py:160
Functionspatial_sampling
Perform spatial sampling on the given video frames. If spatial_idx is -1, perform random scale, random crop, and random flip on the given
slowfast/datasets/utils.py:109
Functionspatial_sampling_label_flip
Perform spatial sampling on the given video frames. If spatial_idx is -1, perform random scale, random crop, and random flip on the given
slowfast/datasets/utils.py:183
Functionspatial_shift_crop_list
Perform left, center, or right crop of the given list of images. Args: size (int): size to crop. image (list): ilist of image
slowfast/datasets/cv2_transform.py:166
Methodstart
Start threads to read and write frames.
slowfast/visualization/demo_loader.py:335
Methodstem
(x)
extract_clip/model.py:139
Functionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
slowfast/utils/distributed.py:138
Functiontest
Perform multi-view testing on the pretrained video model. Args: cfg (CfgNode): configs. Details can be found in slowfast/
tools/test_net.py:143
Functiontokenize
Returns the tokenized representation of given input string(s) Parameters ---------- texts : Union[str, List[str]] An input s
extract_clip/clip.py:197
Functiontopk_accuracies
Computes the top-k accuracy for each k. Args: preds (array): array of predictions. Dimension is N. labels (array): array of l
slowfast/utils/metrics.py:55
Functiontopk_errors
Computes the top-k error for each k. Args: preds (array): array of predictions. Dimension is N. labels (array): array of labe
slowfast/utils/metrics.py:43
Functiontrain
Train a video model for many epochs on train set and evaluate it on val set. Args: cfg (CfgNode): configs. Details can be found in
tools/train_net.py:387
Functiontranslate_x_abs
(img, pixels, **kwargs)
slowfast/datasets/rand_augment.py:96
Functiontranslate_x_rel
(img, pct, **kwargs)
slowfast/datasets/rand_augment.py:80
Functiontranslate_y_abs
(img, pixels, **kwargs)
slowfast/datasets/rand_augment.py:103
Functiontranslate_y_rel
(img, pct, **kwargs)
slowfast/datasets/rand_augment.py:88
Functionuniform_crop
Perform uniform spatial sampling on the images and corresponding boxes. Args: images (tensor): images to perform uniform crop. The di
slowfast/datasets/transform.py:227
Functionuniformerv2_l14
( pretrained=True, use_checkpoint=False, checkpoint_num=[0], t_size=16, dw_reduction=1.5, backbone_dro
slowfast/models/uniformerv2_model.py:404
Functionuniformerv2_l14_336
( pretrained=True, use_checkpoint=False, checkpoint_num=[0], t_size=16, dw_reduction=1.5, backbone_dro
slowfast/models/uniformerv2_model.py:449
Methodupdate
(self, model)
slowfast/utils/ema.py:108
Methodupdate_predictions
Update predictions and labels. Args: preds (tensor): model output predictions. labels (tensor): labels.
slowfast/utils/meters_co.py:687
Methodupdate_stats
Collect the predictions from the current batch and perform on-the-flight summation as ensemble. Args: preds (tens
slowfast/utils/meters.py:297
Methodupdate_stats
Update the current stats. Args: top1_err (float): top1 error rate. top5_err (float): top5 error rate.
slowfast/utils/meters.py:518
Methodupdate_stats
Update the current stats. Args: top1_err (float): top1 error rate. top5_err (float): top5 error rate.
slowfast/utils/meters.py:664
Methodupdate_stats
Update the current stats. Args: preds (tensor): prediction embedding. ori_boxes (tensor): original boxes (x1,
slowfast/utils/meters_co.py:166
Methodupdate_stats
Collect the predictions from the current batch and perform on-the-flight summation as ensemble. Args: preds (tens
slowfast/utils/meters_co.py:297
Methodupdate_stats
Update the current stats. Args: top1_err (float): top1 error rate. top5_err (float): top5 error rate.
slowfast/utils/meters_co.py:524
Methodupdate_stats
Update the current stats. Args: top1_err (float): top1 error rate. top5_err (float): top5 error rate.
slowfast/utils/meters_co.py:673
Functionvisualize
Perform layer weights and activations visualization on the model. Args: cfg (CfgNode): configs. Details can be found in s
tools/visualization.py:256
← previous801–872 of 872, ranked by callers