Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenGVLab/UniFormerV2
/ functions
Functions
872 in github.com/OpenGVLab/UniFormerV2
⨍
Functions
872
◇
Types & classes
124
Function
ioa
Computes pairwise intersection-over-area between box collections. Intersection-over-area (ioa) between two boxes box1 and box2 is defined as
slowfast/utils/ava_evaluation/np_box_ops.py:92
Function
ioa
Computes pairwise intersection-over-area between box collections. Intersection-over-area (ioa) between two masks, mask1 and mask2 is defined as
slowfast/utils/ava_evaluation/np_mask_ops.py:107
Function
iou
Computes pairwise intersection-over-union between box and mask collections. Args: box_mask_list1: BoxMaskList holding N boxes and masks
slowfast/utils/ava_evaluation/np_box_mask_list_ops.py:84
Function
iou
Computes pairwise intersection-over-union between box collections. Args: boxes1: a numpy array with shape [N, 4] holding N boxes. box
slowfast/utils/ava_evaluation/np_box_ops.py:71
Function
iou
Computes pairwise intersection-over-union between box collections. Args: boxlist1: BoxList holding N boxes boxlist2: BoxList holding
slowfast/utils/ava_evaluation/np_box_list_ops.py:71
Function
iou
Computes pairwise intersection-over-union between mask collections. Args: masks1: a numpy array with shape [N, height, width] holding N mas
slowfast/utils/ava_evaluation/np_mask_ops.py:79
Function
is_checkpoint_epoch
Determine if a checkpoint should be saved on current epoch. Args: cfg (CfgNode): configs to save. cur_epoch (int): current nu
slowfast/utils/checkpoint_amp_co.py:99
Function
is_checkpoint_epoch
Determine if a checkpoint should be saved on current epoch. Args: cfg (CfgNode): configs to save. cur_epoch (int): current nu
slowfast/utils/checkpoint.py:95
Function
is_checkpoint_epoch
Determine if a checkpoint should be saved on current epoch. Args: cfg (CfgNode): configs to save. cur_epoch (int): current nu
slowfast/utils/checkpoint_amp_ema.py:99
Function
is_checkpoint_epoch
Determine if a checkpoint should be saved on current epoch. Args: cfg (CfgNode): configs to save. cur_epoch (int): current nu
slowfast/utils/checkpoint_amp.py:99
Function
is_eval_epoch
Determine if the model should be evaluated at the current epoch. Args: cfg (CfgNode): configs. Details can be found in sl
slowfast/utils/misc.py:200
Function
is_master_proc
Determines if the current process is the master process.
slowfast/utils/distributed.py:94
Function
is_root_proc
Determines if the current process is the root process.
slowfast/utils/distributed.py:106
Method
iter_tic
Start to record time.
slowfast/utils/meters.py:348
Method
iter_tic
Start to record time.
slowfast/utils/meters.py:500
Method
iter_tic
Start to record time.
slowfast/utils/meters.py:646
Method
iter_tic
Start to record time.
slowfast/utils/meters_co.py:138
Method
iter_tic
Start to record time.
slowfast/utils/meters_co.py:348
Method
iter_tic
Start to record time.
slowfast/utils/meters_co.py:506
Method
iter_tic
Start to record time.
slowfast/utils/meters_co.py:655
Method
iter_toc
Stop to record time.
slowfast/utils/meters.py:355
Method
iter_toc
Stop to record time.
slowfast/utils/meters.py:507
Method
iter_toc
Stop to record time.
slowfast/utils/meters.py:653
Method
iter_toc
Stop to record time.
slowfast/utils/meters_co.py:145
Method
iter_toc
Stop to record time.
slowfast/utils/meters_co.py:355
Method
iter_toc
Stop to record time.
slowfast/utils/meters_co.py:513
Method
iter_toc
Stop to record time.
slowfast/utils/meters_co.py:662
Method
join
(self)
slowfast/visualization/demo_loader.py:350
Method
last_epoch_time
Get the time for the last epoch.
slowfast/utils/meters_co.py:820
Function
launch_job_co
Run 'func' on one or more GPUs, specified in cfg Args: cfg_list (CfgNode): configs. Details can be found in slowfast/conf
slowfast/utils/misc.py:314
Function
lighting
Perform AlexNet-style PCA jitter on the given image. Args: image (array): list of images to perform lighting jitter. alphastd
slowfast/datasets/cv2_transform.py:556
Function
lighting_jitter
Perform AlexNet-style PCA jitter on the given images. Args: images (tensor): images to perform lighting jitter. Dimension is
slowfast/datasets/transform.py:443
Function
lighting_list
Perform AlexNet-style PCA jitter on the given list of images. Args: images (list): list of images to perform lighting jitter.
slowfast/datasets/cv2_transform.py:273
Function
load
Load a CLIP model Parameters ---------- name : str A model name listed by `clip.available_models()`, or the path to a model check
extract_clip/clip.py:94
Function
load_boxes_and_labels
Loading boxes and labels from csv files. Args: cfg (CfgNode): config. mode (str): 'train', 'val', or 'test' mode. Return
slowfast/datasets/ava_helper.py:68
Function
load_config
Given the arguemnts, load and initialize the configs. Args: args (argument): arguments includes `shard_id`, `num_shards`,
slowfast/utils/parser_co.py:67
Function
load_image_lists
Load image paths and labels from a "frame list". Each line of the frame list contains: `original_vido_id video_id frame_id path labels`
slowfast/datasets/utils.py:306
Function
load_image_lists
Loading image paths from corresponding files. Args: cfg (CfgNode): config. is_train (bool): if it is training dataset or not
slowfast/datasets/ava_helper.py:15
Function
load_pretrained
( model, cfg=None, num_classes=1000, in_chans=3, filter_fn=None, img_size=224, num_f
slowfast/models/helpers.py:48
Function
load_test_checkpoint
Loading checkpoint logic for testing.
slowfast/utils/checkpoint_amp_co.py:526
Function
load_test_checkpoint
Loading checkpoint logic for testing.
slowfast/utils/checkpoint.py:475
Function
load_test_checkpoint
Loading checkpoint logic for testing.
slowfast/utils/checkpoint_amp_ema.py:513
Function
load_test_checkpoint
Loading checkpoint logic for testing.
slowfast/utils/checkpoint_amp.py:510
Function
load_train_checkpoint
Loading checkpoint logic for training.
slowfast/utils/checkpoint_amp_co.py:566
Function
load_train_checkpoint
Loading checkpoint logic for training.
slowfast/utils/checkpoint.py:525
Function
load_train_checkpoint
Loading checkpoint logic for training.
slowfast/utils/checkpoint_amp_ema.py:561
Function
load_train_checkpoint
Loading checkpoint logic for training.
slowfast/utils/checkpoint_amp.py:550
Function
loader_worker_init_fn
Create init function passed to pytorch data loader. Args: dataset (torch.utils.data.Dataset): the given dataset.
slowfast/datasets/utils.py:417
Method
log_epoch_stats
Log the stats of the current epoch. Args: cur_epoch (int): the number of current epoch.
slowfast/utils/meters.py:571
Method
log_epoch_stats
Log the stats of the current epoch. Args: cur_epoch (int): the number of current epoch.
slowfast/utils/meters.py:713
Method
log_epoch_stats
Log the stats of the current epoch. Args: cur_epoch (int): the number of current epoch.
slowfast/utils/meters_co.py:212
Method
log_epoch_stats
Log the stats of the current epoch. Args: cur_iter (int): the number of current iteration.
slowfast/utils/meters_co.py:578
Method
log_epoch_stats
Log the stats of the current epoch. Args: cur_epoch (int): the number of current epoch.
slowfast/utils/meters_co.py:722
Method
log_iter_stats
Log the stats. Args: cur_iter (int): the current iteration of testing.
slowfast/utils/meters.py:332
Method
log_iter_stats
log the stats of the current iteration. Args: cur_epoch (int): the number of current epoch. cur_iter (int): t
slowfast/utils/meters.py:541
Method
log_iter_stats
log the stats of the current iteration. Args: cur_epoch (int): the number of current epoch. cur_iter (int): t
slowfast/utils/meters.py:689
Method
log_iter_stats
Log the stats. Args: cur_epoch (int): the current epoch. cur_iter (int): the current iteration.
slowfast/utils/meters_co.py:86
Method
log_iter_stats
Log the stats. Args: cur_iter (int): the current iteration of testing.
slowfast/utils/meters_co.py:332
Method
log_iter_stats
log the stats of the current iteration. Args: cur_iter (int): the number of current iteration.
slowfast/utils/meters_co.py:547
Method
log_iter_stats
log the stats of the current iteration. Args: cur_epoch (int): the number of current epoch. cur_iter (int): t
slowfast/utils/meters_co.py:698
Function
log_json_stats
Logs json stats. Args: stats (dict): a dictionary of statistical information to log.
slowfast/utils/logging.py:87
Function
log_model_info
Log info, includes number of parameters, gpu usage, gflops and activation count. The model info is computed when the model is in validati
slowfast/utils/misc.py:172
Function
lr_func_cosine
Retrieve the learning rate to specified values at specified epoch with the cosine learning rate schedule. Details can be found in: Ilya L
slowfast/utils/lr_policy.py:30
Function
lr_func_steps_with_relative_lrs
Retrieve the learning rate to specified values at specified epoch with the steps with relative learning rate schedule. Args: cfg
slowfast/utils/lr_policy.py:56
Function
make_checkpoint_dir
Creates the checkpoint directory (if not present already). Args: path_to_job (string): the path to the folder of the current job.
slowfast/utils/checkpoint_amp_co.py:21
Function
make_checkpoint_dir
Creates the checkpoint directory (if not present already). Args: path_to_job (string): the path to the folder of the current job.
slowfast/utils/checkpoint.py:22
Function
make_checkpoint_dir
Creates the checkpoint directory (if not present already). Args: path_to_job (string): the path to the folder of the current job.
slowfast/utils/checkpoint_amp_ema.py:24
Function
make_checkpoint_dir
Creates the checkpoint directory (if not present already). Args: path_to_job (string): the path to the folder of the current job.
slowfast/utils/checkpoint_amp.py:21
Method
median_epoch_time
Calculate the median epoch time among the recorded epochs.
slowfast/utils/meters_co.py:836
Function
merge_pred_gt_boxes
Merge data from precomputed and ground-truth boxes dictionaries. Args: pred_dict (dict): a dict which maps from `frame_idx` to a list
slowfast/visualization/ava_demo_precomputed_boxes.py:362
Function
multi_class_non_max_suppression
Multi-class version of non maximum suppression. This op greedily selects a subset of detection bounding boxes, pruning away boxes that have h
slowfast/utils/ava_evaluation/np_box_mask_list_ops.py:255
Function
multi_class_non_max_suppression
Multi-class version of non maximum suppression. This op greedily selects a subset of detection bounding boxes, pruning away boxes that have h
slowfast/utils/ava_evaluation/np_box_list_ops.py:242
Function
multiple_samples_collate
Collate function for repeated augmentation. Each instance in the batch has more than one sample. Args: batch (tuple or list): dat
slowfast/datasets/loader.py:20
Method
no_weight_decay
(self)
slowfast/models/video_model_builder.py:944
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/kinetics.py:407
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/mit.py:415
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/ava_dataset.py:106
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/kinetics_sparse.py:416
Method
num_videos
Returns: (int): the number of clips in total in the dataset.
slowfast/datasets/ptv_datasets.py:83
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/anet.py:437
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/charades.py:253
Method
num_videos
Returns: (int): the number of videos in the dataset.
slowfast/datasets/sth.py:375
Function
parse_args
Parse the following arguments for a default parser for PySlowFast users. Args: shard_id (int): shard id for the current machine. Star
slowfast/utils/parser_co.py:13
Function
plot_confusion_matrix
A function to create a colored and labeled confusion matrix matplotlib figure given true labels and preds. Args: cmtx (ndarray):
slowfast/visualization/utils.py:48
Function
plot_input
Plot the input tensor with the optional bounding box and save it to disk. Args: tensor (tensor): a tensor with shape of `NxCxHxW`.
slowfast/utils/misc.py:224
Function
plot_topk_histogram
Plot histogram of top-k value from the given array. Args: tag (str): histogram title. array (tensor): a tensor to draw top k
slowfast/visualization/utils.py:92
Function
posterize
(img, bits_to_keep, **__)
slowfast/datasets/rand_augment.py:175
Function
print_pass
(*objects, sep=" ", end="\n", file=sys.stdout, flush=False)
slowfast/utils/logging.py:24
Function
process_charades_label
Process the video label for Charades dataset. Use video-level label for training mode, otherwise use clip-level label. Then convert the label
slowfast/datasets/ptv_datasets.py:274
Function
prune_non_overlapping_boxes
Prunes the boxes in boxlist1 that overlap less than thresh with boxlist2. For each box in boxlist1, we want its IOA to be more than minoverlap wi
slowfast/utils/ava_evaluation/np_box_list_ops.py:385
Function
prune_non_overlapping_masks
Prunes the boxes in list1 that overlap less than thresh with list2. For each mask in box_mask_list1, we want its IOA to be more than minoverlap
slowfast/utils/ava_evaluation/np_box_mask_list_ops.py:337
Function
prune_outside_window
Prunes bounding boxes that fall outside a given window. This function prunes bounding boxes that even partially fall outside the given window
slowfast/utils/ava_evaluation/np_box_list_ops.py:410
Method
put
Put task into task queue for prediction and visualization. Args: task (TaskInfo object): task object that contain
slowfast/visualization/async_predictor.py:254
Method
put_fn
Grabbing frames from VideoCapture.
slowfast/visualization/demo_loader.py:229
Function
random_crop
Perform random spatial crop on the given images and corresponding boxes. Args: images (tensor): images to perform random crop. The di
slowfast/datasets/transform.py:117
Function
random_crop_list
Perform random crop on a list of images. Args: images (list): list of images to perform random crop. size (int): size to crop
slowfast/datasets/cv2_transform.py:394
Function
random_resized_crop
Crop the given images to random size and aspect ratio. A crop of random size (default: of 0.08 to 1.0) of the original size and a random aspe
slowfast/datasets/transform.py:577
Function
random_resized_crop_with_shift
This is similar to random_resized_crop. However, it samples two different boxes (for cropping) for the first and last frame. It then linearly
slowfast/datasets/transform.py:612
Function
random_scale_jitter
Perform ResNet style random scale jittering: randomly select the scale from [1/max_size, 1/min_size]. Args: image (array): im
slowfast/datasets/cv2_transform.py:477
Function
random_scale_jitter_list
Perform ResNet style random scale jittering on a list of image: randomly select the scale from [1/max_size, 1/min_size]. Note that all th
slowfast/datasets/cv2_transform.py:495
← previous
next →
701–800 of 872, ranked by callers