SYMBOL INDEX (1723 symbols across 157 files)

FILE: build/lib/slowfast/config/custom_config.py
  function add_custom_config (line 7) | def add_custom_config(_C):

FILE: build/lib/slowfast/config/defaults.py
  function assert_and_infer_cfg (line 990) | def assert_and_infer_cfg(cfg):
  function get_cfg (line 1018) | def get_cfg():

FILE: build/lib/slowfast/datasets/ava_dataset.py
  class Ava (line 18) | class Ava(torch.utils.data.Dataset):
    method __init__ (line 23) | def __init__(self, cfg, split):
    method _load_data (line 49) | def _load_data(self, cfg):
    method print_summary (line 87) | def print_summary(self):
    method __len__ (line 98) | def __len__(self):
    method num_videos (line 106) | def num_videos(self):
    method _images_and_boxes_preprocessing_cv2 (line 113) | def _images_and_boxes_preprocessing_cv2(self, imgs, boxes):
    method _images_and_boxes_preprocessing (line 245) | def _images_and_boxes_preprocessing(self, imgs, boxes):
    method __getitem__ (line 352) | def __getitem__(self, idx):

FILE: build/lib/slowfast/datasets/ava_helper.py
  function load_image_lists (line 16) | def load_image_lists(cfg, is_train):
  function load_boxes_and_labels (line 69) | def load_boxes_and_labels(cfg, mode):
  function get_keyframe_data (line 117) | def get_keyframe_data(boxes_and_labels):
  function get_num_boxes_used (line 163) | def get_num_boxes_used(keyframe_indices, keyframe_boxes_and_labels):
  function parse_bboxes_file (line 182) | def parse_bboxes_file(

FILE: build/lib/slowfast/datasets/build.py
  function build_dataset (line 15) | def build_dataset(dataset_name, cfg, split):

FILE: build/lib/slowfast/datasets/charades.py
  class Charades (line 20) | class Charades(torch.utils.data.Dataset):
    method __init__ (line 31) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 74) | def _construct_loader(self):
    method get_seq_frames (line 113) | def get_seq_frames(self, index):
    method __getitem__ (line 154) | def __getitem__(self, index):
    method __len__ (line 249) | def __len__(self):
    method num_videos (line 257) | def num_videos(self):

FILE: build/lib/slowfast/datasets/cv2_transform.py
  function clip_boxes_to_image (line 9) | def clip_boxes_to_image(boxes, height, width):
  function random_short_side_scale_jitter_list (line 29) | def random_short_side_scale_jitter_list(images, min_size, max_size, boxe...
  function scale (line 77) | def scale(size, image):
  function scale_boxes (line 106) | def scale_boxes(size, boxes, height, width):
  function horizontal_flip_list (line 134) | def horizontal_flip_list(prob, images, order="CHW", boxes=None):
  function spatial_shift_crop_list (line 166) | def spatial_shift_crop_list(size, images, spatial_shift_pos, boxes=None):
  function CHW2HWC (line 216) | def CHW2HWC(image):
  function HWC2CHW (line 228) | def HWC2CHW(image):
  function color_jitter_list (line 240) | def color_jitter_list(
  function lighting_list (line 273) | def lighting_list(imgs, alphastd, eigval, eigvec, alpha=None):
  function color_normalization (line 302) | def color_normalization(image, mean, stddev):
  function pad_image (line 319) | def pad_image(image, pad_size, order="CHW"):
  function horizontal_flip (line 344) | def horizontal_flip(prob, image, order="CHW"):
  function flip_boxes (line 365) | def flip_boxes(boxes, im_width):
  function crop_boxes (line 381) | def crop_boxes(boxes, x_offset, y_offset):
  function random_crop_list (line 394) | def random_crop_list(images, size, pad_size=0, order="CHW", boxes=None):
  function center_crop (line 458) | def center_crop(size, image):
  function random_scale_jitter (line 477) | def random_scale_jitter(image, min_size, max_size):
  function random_scale_jitter_list (line 495) | def random_scale_jitter_list(images, min_size, max_size):
  function random_sized_crop (line 513) | def random_sized_crop(image, size, area_frac=0.08):
  function lighting (line 556) | def lighting(img, alphastd, eigval, eigvec):
  function random_sized_crop_list (line 582) | def random_sized_crop_list(images, size, crop_area_fraction=0.08):
  function blend (line 632) | def blend(image1, image2, alpha):
  function grayscale (line 636) | def grayscale(image):
  function saturation (line 654) | def saturation(var, image):
  function brightness (line 668) | def brightness(var, image):
  function contrast (line 682) | def contrast(var, image):
  function saturation_list (line 697) | def saturation_list(var, images):
  function brightness_list (line 715) | def brightness_list(var, images):
  function contrast_list (line 733) | def contrast_list(var, images):
  function color_jitter (line 752) | def color_jitter(image, img_brightness=0, img_contrast=0, img_saturation...
  function revert_scaled_boxes (line 783) | def revert_scaled_boxes(size, boxes, img_height, img_width):

FILE: build/lib/slowfast/datasets/decoder.py
  function temporal_sampling (line 11) | def temporal_sampling(frames, start_idx, end_idx, num_samples):
  function get_start_end_idx (line 31) | def get_start_end_idx(
  function pyav_decode_stream (line 72) | def pyav_decode_stream(
  function torchvision_decode (line 114) | def torchvision_decode(
  function pyav_decode (line 229) | def pyav_decode(
  function decode (line 305) | def decode(

FILE: build/lib/slowfast/datasets/imagenet.py
  class Imagenet (line 26) | class Imagenet(torch.utils.data.Dataset):
    method __init__ (line 29) | def __init__(self, cfg, mode, num_retries=10):
    method _load_imdb (line 45) | def _load_imdb(self):
    method _construct_imdb (line 53) | def _construct_imdb(self):
    method load_image (line 76) | def load_image(self, im_path):
    method _prepare_im_res (line 86) | def _prepare_im_res(self, im_path):
    method _prepare_im_tf (line 123) | def _prepare_im_tf(self, im_path):
    method __load__ (line 162) | def __load__(self, index):
    method __getitem__ (line 183) | def __getitem__(self, index):
    method __len__ (line 202) | def __len__(self):

FILE: build/lib/slowfast/datasets/kinetics.py
  class Kinetics (line 24) | class Kinetics(torch.utils.data.Dataset):
    method __init__ (line 35) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 88) | def _construct_loader(self):
    method __getitem__ (line 129) | def __getitem__(self, index):
    method _aug_frame (line 320) | def _aug_frame(
    method _frame_to_list_img (line 385) | def _frame_to_list_img(self, frames):
    method _list_img_to_frames (line 391) | def _list_img_to_frames(self, img_list):
    method __len__ (line 395) | def __len__(self):
    method num_videos (line 403) | def num_videos(self):

FILE: build/lib/slowfast/datasets/loader.py
  function multiple_samples_collate (line 20) | def multiple_samples_collate(batch, fold=False):
  function detection_collate (line 48) | def detection_collate(batch):
  function construct_loader (line 85) | def construct_loader(cfg, split, is_precise_bn=False):
  function shuffle_dataset (line 173) | def shuffle_dataset(loader, cur_epoch):

FILE: build/lib/slowfast/datasets/mixup.py
  function convert_to_one_hot (line 22) | def convert_to_one_hot(targets, num_classes, on_value=1.0, off_value=0.0):
  function mixup_target (line 40) | def mixup_target(target, num_classes, lam=1.0, smoothing=0.0):
  function rand_bbox (line 67) | def rand_bbox(img_shape, lam, margin=0.0, count=None):
  function get_cutmix_bbox (line 90) | def get_cutmix_bbox(img_shape, lam, correct_lam=True, count=None):
  class MixUp (line 109) | class MixUp:
    method __init__ (line 117) | def __init__(
    method _get_mixup_params (line 148) | def _get_mixup_params(self):
    method _mix_batch (line 167) | def _mix_batch(self, x):
    method __call__ (line 183) | def __call__(self, x, target):

FILE: build/lib/slowfast/datasets/multigrid_helper.py
  class ShortCycleBatchSampler (line 19) | class ShortCycleBatchSampler(Sampler):
    method __init__ (line 26) | def __init__(self, sampler, batch_size, drop_last, cfg):
    method __iter__ (line 68) | def __iter__(self):
    method __len__ (line 82) | def __len__(self):

FILE: build/lib/slowfast/datasets/ptv_datasets.py
  class PTVDatasetWrapper (line 42) | class PTVDatasetWrapper(torch.utils.data.IterableDataset):
    method __init__ (line 47) | def __init__(self, num_videos, clips_per_video, crops_per_clip, dataset):
    method __next__ (line 61) | def __next__(self):
    method sampler (line 68) | def sampler(self):
    method __len__ (line 75) | def __len__(self):
    method num_videos (line 83) | def num_videos(self):
    method __iter__ (line 90) | def __iter__(self):
  class PackPathway (line 94) | class PackPathway(torch.nn.Module):
    method __init__ (line 100) | def __init__(self, cfg):
    method forward (line 104) | def forward(self, x: torch.Tensor):
  class DictToTuple (line 108) | class DictToTuple(torch.nn.Module):
    method __init__ (line 114) | def __init__(self, num_clips, num_crops):
    method forward (line 119) | def forward(self, x: Dict[str, torch.Tensor]):
  function div255 (line 129) | def div255(x):
  function Ptvkinetics (line 143) | def Ptvkinetics(cfg, mode):
  function process_charades_label (line 275) | def process_charades_label(x, mode, num_classes):
  function rgb2bgr (line 298) | def rgb2bgr(x):
  function Ptvcharades (line 312) | def Ptvcharades(cfg, mode):
  function Ptvssv2 (line 455) | def Ptvssv2(cfg, mode):

FILE: build/lib/slowfast/datasets/rand_augment.py
  function _interpolation (line 52) | def _interpolation(kwargs):
  function _check_args_tf (line 60) | def _check_args_tf(kwargs):
  function shear_x (line 66) | def shear_x(img, factor, **kwargs):
  function shear_y (line 73) | def shear_y(img, factor, **kwargs):
  function translate_x_rel (line 80) | def translate_x_rel(img, pct, **kwargs):
  function translate_y_rel (line 88) | def translate_y_rel(img, pct, **kwargs):
  function translate_x_abs (line 96) | def translate_x_abs(img, pixels, **kwargs):
  function translate_y_abs (line 103) | def translate_y_abs(img, pixels, **kwargs):
  function rotate (line 110) | def rotate(img, degrees, **kwargs):
  function auto_contrast (line 144) | def auto_contrast(img, **__):
  function invert (line 148) | def invert(img, **__):
  function equalize (line 152) | def equalize(img, **__):
  function solarize (line 156) | def solarize(img, thresh, **__):
  function solarize_add (line 160) | def solarize_add(img, add, thresh=128, **__):
  function posterize (line 175) | def posterize(img, bits_to_keep, **__):
  function contrast (line 181) | def contrast(img, factor, **__):
  function color (line 185) | def color(img, factor, **__):
  function brightness (line 189) | def brightness(img, factor, **__):
  function sharpness (line 193) | def sharpness(img, factor, **__):
  function _randomly_negate (line 197) | def _randomly_negate(v):
  function _rotate_level_to_arg (line 202) | def _rotate_level_to_arg(level, _hparams):
  function _enhance_level_to_arg (line 209) | def _enhance_level_to_arg(level, _hparams):
  function _enhance_increasing_level_to_arg (line 214) | def _enhance_increasing_level_to_arg(level, _hparams):
  function _shear_level_to_arg (line 222) | def _shear_level_to_arg(level, _hparams):
  function _translate_abs_level_to_arg (line 229) | def _translate_abs_level_to_arg(level, hparams):
  function _translate_rel_level_to_arg (line 236) | def _translate_rel_level_to_arg(level, hparams):
  function _posterize_level_to_arg (line 244) | def _posterize_level_to_arg(level, _hparams):
  function _posterize_increasing_level_to_arg (line 251) | def _posterize_increasing_level_to_arg(level, hparams):
  function _posterize_original_level_to_arg (line 258) | def _posterize_original_level_to_arg(level, _hparams):
  function _solarize_level_to_arg (line 265) | def _solarize_level_to_arg(level, _hparams):
  function _solarize_increasing_level_to_arg (line 271) | def _solarize_increasing_level_to_arg(level, _hparams):
  function _solarize_add_level_to_arg (line 277) | def _solarize_add_level_to_arg(level, _hparams):
  class AugmentOp (line 339) | class AugmentOp:
    method __init__ (line 344) | def __init__(self, name, prob=0.5, magnitude=10, hparams=None):
    method __call__ (line 366) | def __call__(self, img_list):
  function _select_rand_weights (line 446) | def _select_rand_weights(weight_idx=0, transforms=None):
  function rand_augment_ops (line 455) | def rand_augment_ops(magnitude=10, hparams=None, transforms=None):
  class RandAugment (line 464) | class RandAugment:
    method __init__ (line 465) | def __init__(self, ops, num_layers=2, choice_weights=None):
    method __call__ (line 470) | def __call__(self, img):
  function rand_augment_transform (line 483) | def rand_augment_transform(config_str, hparams):

FILE: build/lib/slowfast/datasets/random_erasing.py
  function _get_pixels (line 18) | def _get_pixels(
  class RandomErasing (line 34) | class RandomErasing:
    method __init__ (line 53) | def __init__(
    method _erase (line 87) | def _erase(self, img, chan, img_h, img_w, dtype):
    method _erase_cube (line 116) | def _erase_cube(
    method __call__ (line 158) | def __call__(self, input):

FILE: build/lib/slowfast/datasets/ssv2.py
  class Ssv2 (line 22) | class Ssv2(torch.utils.data.Dataset):
    method __init__ (line 33) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 73) | def _construct_loader(self):
    method get_seq_frames (line 159) | def get_seq_frames(self, index):
    method __getitem__ (line 182) | def __getitem__(self, index):
    method __len__ (line 272) | def __len__(self):
    method num_videos (line 280) | def num_videos(self):

FILE: build/lib/slowfast/datasets/sth.py
  class Sth (line 26) | class Sth(torch.utils.data.Dataset):
    method __init__ (line 37) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 86) | def _construct_loader(self):
    method get_seq_frames (line 141) | def get_seq_frames(self, index, temporal_sample_index):
    method __getitem__ (line 170) | def __getitem__(self, index):
    method _aug_frame (line 307) | def _aug_frame(
    method _frame_to_list_img (line 372) | def _frame_to_list_img(self, frames):
    method _list_img_to_frames (line 378) | def _list_img_to_frames(self, img_list):
    method __len__ (line 382) | def __len__(self):
    method num_videos (line 390) | def num_videos(self):

FILE: build/lib/slowfast/datasets/transform.py
  function _pil_interp (line 33) | def _pil_interp(method):
  function random_short_side_scale_jitter (line 47) | def random_short_side_scale_jitter(
  function crop_boxes (line 104) | def crop_boxes(boxes, x_offset, y_offset):
  function random_crop (line 123) | def random_crop(images, size, boxes=None):
  function horizontal_flip (line 159) | def horizontal_flip(prob, images, boxes=None):
  function uniform_crop (line 194) | def uniform_crop(images, size, spatial_idx, boxes=None, scale_size=None):
  function clip_boxes_to_image (line 257) | def clip_boxes_to_image(boxes, height, width):
  function blend (line 279) | def blend(images1, images2, alpha):
  function grayscale (line 295) | def grayscale(images):
  function color_jitter (line 317) | def color_jitter(images, img_brightness=0, img_contrast=0, img_saturatio...
  function brightness_jitter (line 352) | def brightness_jitter(var, images):
  function contrast_jitter (line 371) | def contrast_jitter(var, images):
  function saturation_jitter (line 391) | def saturation_jitter(var, images):
  function lighting_jitter (line 410) | def lighting_jitter(images, alphastd, eigval, eigvec):
  function color_normalization (line 458) | def color_normalization(images, mean, stddev):
  function _get_param_spatial_crop (line 502) | def _get_param_spatial_crop(
  function random_resized_crop (line 544) | def random_resized_crop(
  function random_resized_crop_with_shift (line 579) | def random_resized_crop_with_shift(
  function create_random_augment (line 624) | def create_random_augment(
  function random_sized_crop_img (line 660) | def random_sized_crop_img(
  class RandomResizedCropAndInterpolation (line 695) | class RandomResizedCropAndInterpolation:
    method __init__ (line 708) | def __init__(
    method get_params (line 730) | def get_params(img, scale, ratio):
    method __call__ (line 770) | def __call__(self, img):
    method __repr__ (line 784) | def __repr__(self):
  function transforms_imagenet_train (line 802) | def transforms_imagenet_train(
  function temporal_difference (line 903) | def temporal_difference(
  function color_jitter_video_ssl (line 929) | def color_jitter_video_ssl(
  function augment_raw_frames (line 980) | def augment_raw_frames(frames, time_diff_prob=0.0, gaussian_prob=0.0):
  class GaussianBlur (line 1003) | class GaussianBlur(object):
    method __init__ (line 1006) | def __init__(self, sigma=[0.1, 2.0]):
    method __call__ (line 1009) | def __call__(self, x):
  class GaussianBlurVideo (line 1018) | class GaussianBlurVideo(object):
    method __init__ (line 1019) | def __init__(
    method __call__ (line 1025) | def __call__(self, frames):

FILE: build/lib/slowfast/datasets/utils.py
  function retry_load_images (line 20) | def retry_load_images(image_paths, retry=10, backend="pytorch"):
  function get_sequence (line 51) | def get_sequence(center_idx, half_len, sample_rate, num_frames):
  function pack_pathway_output (line 74) | def pack_pathway_output(cfg, frames):
  function spatial_sampling (line 110) | def spatial_sampling(
  function as_binary_vector (line 184) | def as_binary_vector(labels, num_classes):
  function aggregate_labels (line 200) | def aggregate_labels(label_list):
  function convert_to_video_level_labels (line 215) | def convert_to_video_level_labels(labels):
  function load_image_lists (line 231) | def load_image_lists(frame_list_file, prefix="", return_list=False):
  function tensor_normalize (line 276) | def tensor_normalize(tensor, mean, std, func=None):
  function get_random_sampling_rate (line 298) | def get_random_sampling_rate(long_cycle_sampling_rate, sampling_rate):
  function revert_tensor_normalize (line 310) | def revert_tensor_normalize(tensor, mean, std):
  function create_sampler (line 327) | def create_sampler(dataset, shuffle, cfg):
  function loader_worker_init_fn (line 344) | def loader_worker_init_fn(dataset):

FILE: build/lib/slowfast/datasets/video_container.py
  function get_video_container (line 9) | def get_video_container(path_to_vid, multi_thread_decode=False, backend=...

FILE: build/lib/slowfast/models/attention.py
  function attention_pool (line 14) | def attention_pool(tensor, pool, thw_shape, has_cls_embed=True, norm=None):
  function get_rel_pos (line 51) | def get_rel_pos(rel_pos, d):
  function cal_rel_pos_spatial (line 67) | def cal_rel_pos_spatial(
  function cal_rel_pos_temporal (line 120) | def cal_rel_pos_temporal(attn, q, has_cls_embed, q_shape, k_shape, rel_p...
  class MultiScaleAttention (line 162) | class MultiScaleAttention(nn.Module):
    method __init__ (line 163) | def __init__(
    method forward (line 307) | def forward(self, x, thw_shape):
  class MultiScaleBlock (line 445) | class MultiScaleBlock(nn.Module):
    method __init__ (line 446) | def __init__(
    method forward (line 533) | def forward(self, x, thw_shape):

FILE: build/lib/slowfast/models/batchnorm_helper.py
  function get_norm (line 16) | def get_norm(cfg):
  class SubBatchNorm3d (line 40) | class SubBatchNorm3d(nn.Module):
    method __init__ (line 51) | def __init__(self, num_splits, **args):
    method _get_aggregated_mean_std (line 72) | def _get_aggregated_mean_std(self, means, stds, n):
    method aggregate_stats (line 87) | def aggregate_stats(self):
    method forward (line 101) | def forward(self, x):

FILE: build/lib/slowfast/models/build.py
  function build_model (line 18) | def build_model(cfg, gpu_id=None):

FILE: build/lib/slowfast/models/common.py
  class Mlp (line 7) | class Mlp(nn.Module):
    method __init__ (line 8) | def __init__(
    method forward (line 26) | def forward(self, x):
  class Permute (line 37) | class Permute(nn.Module):
    method __init__ (line 38) | def __init__(self, dims):
    method forward (line 42) | def forward(self, x):
  function drop_path (line 46) | def drop_path(x, drop_prob: float = 0.0, training: bool = False):
  class DropPath (line 62) | class DropPath(nn.Module):
    method __init__ (line 65) | def __init__(self, drop_prob=None):
    method forward (line 69) | def forward(self, x):

FILE: build/lib/slowfast/models/contrastive.py
  class ContrastiveModel (line 34) | class ContrastiveModel(nn.Module):
    method __init__ (line 39) | def __init__(self, cfg):
    method knn_mem_update (line 134) | def knn_mem_update(self, q_knn, index):
    method init_knn_labels (line 145) | def init_knn_labels(self, train_loader):
    method _update_history (line 161) | def _update_history(self):
    method _batch_shuffle (line 177) | def _batch_shuffle(self, x):
    method _batch_unshuffle (line 219) | def _batch_unshuffle(self, x, idx_restore):
    method eval_knn (line 235) | def eval_knn(self, q_knn, knn_k=200):
    method sim_loss (line 245) | def sim_loss(self, q, k):
    method momentum_anneal_cosine (line 254) | def momentum_anneal_cosine(self, epoch_exact):
    method _dequeue_and_enqueue (line 266) | def _dequeue_and_enqueue(self, keys, extra_keys=None):
    method batch_clips (line 297) | def batch_clips(self, clips):
    method compute_key_feat (line 311) | def compute_key_feat(
    method forward (line 375) | def forward(
    method _simclr_precompute_pos_neg_mask_multi (line 808) | def _simclr_precompute_pos_neg_mask_multi(self):
    method run_swav_encoder_q (line 850) | def run_swav_encoder_q(self, im):
    method get_code (line 858) | def get_code(self, out):
    method run_swav_orig_encoder_q (line 867) | def run_swav_orig_encoder_q(self, x):
    method sinkhorn (line 875) | def sinkhorn(self, Q, iters):
    method distributed_sinkhorn (line 891) | def distributed_sinkhorn(self, Q, nmb_iters):
    method KLDivLoss (line 914) | def KLDivLoss(self, out, code):
  function l2_loss (line 921) | def l2_loss(x, y):
  class Normalize (line 925) | class Normalize(nn.Module):
    method __init__ (line 926) | def __init__(self, power=2, dim=1):
    method forward (line 931) | def forward(self, x):
  class Memory (line 939) | class Memory(nn.Module):
    method __init__ (line 940) | def __init__(self, length, duration, dim, cfg):
    method resize (line 955) | def resize(self, length, duration, dim):
    method get (line 968) | def get(self, ind, time, interp=False):
    method update (line 993) | def update(self, mem, momentum, ind, time, interp=False):
    method forward (line 1040) | def forward(self, inputs):
  class Memory1D (line 1044) | class Memory1D(nn.Module):
    method __init__ (line 1045) | def __init__(self, length, duration, dim, cfg):
    method get (line 1059) | def get(self, ind, time, interp=False):
    method update (line 1071) | def update(self, mem, momentum, ind, time, interp=False):
  function cancel_swav_gradients (line 1085) | def cancel_swav_gradients(model, cfg, epoch_exact):

FILE: build/lib/slowfast/models/head_helper.py
  class ResNetRoIHead (line 16) | class ResNetRoIHead(nn.Module):
    method __init__ (line 21) | def __init__(
    method forward (line 114) | def forward(self, inputs, bboxes):
  class MLPHead (line 146) | class MLPHead(nn.Module):
    method __init__ (line 147) | def __init__(
    method forward (line 188) | def forward(self, x):
  class ResNetBasicHead (line 197) | class ResNetBasicHead(nn.Module):
    method __init__ (line 206) | def __init__(
    method forward (line 309) | def forward(self, inputs):
  class X3DHead (line 357) | class X3DHead(nn.Module):
    method __init__ (line 366) | def __init__(
    method _construct_head (line 416) | def _construct_head(self, dim_in, dim_inner, dim_out, norm_module):
    method forward (line 467) | def forward(self, inputs):
  class TransformerBasicHead (line 497) | class TransformerBasicHead(nn.Module):
    method __init__ (line 502) | def __init__(
    method forward (line 556) | def forward(self, x):

FILE: build/lib/slowfast/models/losses.py
  class ContrastiveLoss (line 16) | class ContrastiveLoss(nn.Module):
    method __init__ (line 17) | def __init__(self, reduction="mean"):
    method forward (line 21) | def forward(self, inputs, dummy_labels=None):
  function get_loss_func (line 38) | def get_loss_func(loss_name):

FILE: build/lib/slowfast/models/morphmlp.py
  function conv_3xnxn (line 12) | def conv_3xnxn(inp, oup, kernel_size=3, stride=3):
  function conv_1xnxn (line 16) | def conv_1xnxn(inp, oup, kernel_size=3, stride=3):
  class Mlp (line 19) | class Mlp(nn.Module):
    method __init__ (line 20) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 29) | def forward(self, x):
  class MorphFC_S2 (line 38) | class MorphFC_S2(nn.Module):
    method __init__ (line 39) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 51) | def forward(self, x):
  class MorphFC_S (line 115) | class MorphFC_S(nn.Module):
    method __init__ (line 116) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 130) | def forward(self, x):
  class MorphFC_T (line 156) | class MorphFC_T(nn.Module):
    method __init__ (line 157) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 168) | def forward(self, x):
  class PermutatorBlock (line 185) | class PermutatorBlock(nn.Module):
    method __init__ (line 186) | def __init__(self, dim, segment_dim, mlp_ratio=4., qkv_bias=False, qk_...
    method forward (line 203) | def forward(self, x):
  class PatchEmbed (line 210) | class PatchEmbed(nn.Module):
    method __init__ (line 214) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=...
    method forward (line 222) | def forward(self, x):
  class Downsample (line 230) | class Downsample(nn.Module):
    method __init__ (line 234) | def __init__(self, in_embed_dim, out_embed_dim, patch_size):
    method forward (line 239) | def forward(self, x):
  class MorphMLP (line 246) | class MorphMLP(nn.Module):
    method __init__ (line 250) | def __init__(self, cfg):
    method _init_weights (line 350) | def _init_weights(self, m):
    method get_pretrained_model (line 359) | def get_pretrained_model(self, cfg):
    method forward_features (line 369) | def forward_features(self, x):
    method forward (line 389) | def forward(self, x):

FILE: build/lib/slowfast/models/morphmlp_32.py
  function conv_3xnxn (line 10) | def conv_3xnxn(inp, oup, kernel_size=3, stride=3):
  function conv_1xnxn (line 14) | def conv_1xnxn(inp, oup, kernel_size=3, stride=3):
  class Mlp (line 17) | class Mlp(nn.Module):
    method __init__ (line 18) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 27) | def forward(self, x):
  class MorphFC_S2 (line 36) | class MorphFC_S2(nn.Module):
    method __init__ (line 37) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 49) | def forward(self, x):
  class MorphFC_S (line 113) | class MorphFC_S(nn.Module):
    method __init__ (line 114) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 128) | def forward(self, x):
  class MorphFC_T (line 154) | class MorphFC_T(nn.Module):
    method __init__ (line 155) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 169) | def forward(self, x):
  class PermutatorBlock (line 186) | class PermutatorBlock(nn.Module):
    method __init__ (line 187) | def __init__(self, dim, segment_dim, mlp_ratio=4., qkv_bias=False, qk_...
    method forward (line 204) | def forward(self, x):
  class PatchEmbed (line 211) | class PatchEmbed(nn.Module):
    method __init__ (line 215) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=...
    method forward (line 223) | def forward(self, x):
  class Downsample (line 231) | class Downsample(nn.Module):
    method __init__ (line 235) | def __init__(self, in_embed_dim, out_embed_dim, patch_size):
    method forward (line 240) | def forward(self, x):
  class MorphMLP_32 (line 247) | class MorphMLP_32(nn.Module):
    method __init__ (line 251) | def __init__(self, cfg):
    method _init_weights (line 348) | def _init_weights(self, m):
    method get_pretrained_model (line 357) | def get_pretrained_model(self, cfg):
    method forward_features (line 367) | def forward_features(self, x):
    method forward (line 387) | def forward(self, x):

FILE: build/lib/slowfast/models/nonlocal_helper.py
  class Nonlocal (line 10) | class Nonlocal(nn.Module):
    method __init__ (line 20) | def __init__(
    method _construct_nonlocal (line 67) | def _construct_nonlocal(
    method forward (line 105) | def forward(self, x):

FILE: build/lib/slowfast/models/operators.py
  class SE (line 11) | class SE(nn.Module):
    method _round_width (line 14) | def _round_width(self, width, multiplier, min_width=8, divisor=8):
    method __init__ (line 35) | def __init__(self, dim_in, ratio, relu_act=True):
    method forward (line 53) | def forward(self, x):

FILE: build/lib/slowfast/models/optimizer.py
  function construct_optimizer (line 11) | def construct_optimizer(model, cfg):
  function get_epoch_lr (line 101) | def get_epoch_lr(cur_epoch, cfg):
  function set_lr (line 112) | def set_lr(optimizer, new_lr):

FILE: build/lib/slowfast/models/ptv_model_builder.py
  function get_head_act (line 37) | def get_head_act(act_func):
  class PTVResNet (line 59) | class PTVResNet(nn.Module):
    method __init__ (line 64) | def __init__(self, cfg):
    method _construct_network (line 90) | def _construct_network(self, cfg):
    method forward (line 209) | def forward(self, x, bboxes=None):
  class PTVSlowFast (line 225) | class PTVSlowFast(nn.Module):
    method __init__ (line 226) | def __init__(self, cfg):
    method _construct_network (line 247) | def _construct_network(self, cfg):
    method forward (line 410) | def forward(self, x, bboxes=None):
  class PTVX3D (line 425) | class PTVX3D(nn.Module):
    method __init__ (line 426) | def __init__(self, cfg):
    method _construct_network (line 449) | def _construct_network(self, cfg):
    method forward (line 505) | def forward(self, x, bboxes=None):
  class PTVCSN (line 518) | class PTVCSN(nn.Module):
    method __init__ (line 523) | def __init__(self, cfg):
    method _construct_network (line 540) | def _construct_network(self, cfg):
    method forward (line 590) | def forward(self, x, bboxes=None):
  class PTVR2plus1D (line 603) | class PTVR2plus1D(nn.Module):
    method __init__ (line 608) | def __init__(self, cfg):
    method _construct_network (line 625) | def _construct_network(self, cfg):
    method forward (line 691) | def forward(self, x, bboxes=None):
  class PTVMViT (line 704) | class PTVMViT(nn.Module):
    method __init__ (line 709) | def __init__(self, cfg):
    method _construct_network (line 726) | def _construct_network(self, cfg):
    method forward (line 770) | def forward(self, x, bboxes=None):

FILE: build/lib/slowfast/models/resnet_helper.py
  function get_trans_func (line 14) | def get_trans_func(name):
  class BasicTransform (line 29) | class BasicTransform(nn.Module):
    method __init__ (line 34) | def __init__(
    method _construct (line 77) | def _construct(self, dim_in, dim_out, stride, dilation, norm_module):
    method forward (line 110) | def forward(self, x):
  class X3DTransform (line 120) | class X3DTransform(nn.Module):
    method __init__ (line 127) | def __init__(
    method _construct (line 190) | def _construct(
    method forward (line 255) | def forward(self, x):
  class BottleneckTransform (line 261) | class BottleneckTransform(nn.Module):
    method __init__ (line 267) | def __init__(
    method _construct (line 321) | def _construct(
    method forward (line 379) | def forward(self, x):
  class ResBlock (line 397) | class ResBlock(nn.Module):
    method __init__ (line 402) | def __init__(
    method _construct (line 470) | def _construct(
    method forward (line 514) | def forward(self, x):
  class ResStage (line 526) | class ResStage(nn.Module):
    method __init__ (line 537) | def __init__(
    method _construct (line 652) | def _construct(
    method forward (line 701) | def forward(self, inputs):

FILE: build/lib/slowfast/models/stem_helper.py
  function get_stem_func (line 9) | def get_stem_func(name):
  class VideoModelStem (line 20) | class VideoModelStem(nn.Module):
    method __init__ (line 26) | def __init__(
    method _construct_stem (line 98) | def _construct_stem(self, dim_in, dim_out, norm_module, stem_func_name):
    method forward (line 115) | def forward(self, x):
  class ResNetBasicStem (line 127) | class ResNetBasicStem(nn.Module):
    method __init__ (line 134) | def __init__(
    method _construct_stem (line 181) | def _construct_stem(self, dim_in, dim_out, norm_module):
    method forward (line 198) | def forward(self, x):
  class X3DStem (line 206) | class X3DStem(nn.Module):
    method __init__ (line 213) | def __init__(
    method _construct_stem (line 260) | def _construct_stem(self, dim_in, dim_out, norm_module):
    method forward (line 284) | def forward(self, x):
  class PatchEmbed (line 292) | class PatchEmbed(nn.Module):
    method __init__ (line 297) | def __init__(
    method forward (line 319) | def forward(self, x):

FILE: build/lib/slowfast/models/utils.py
  function round_width (line 8) | def round_width(width, multiplier, min_width=1, divisor=1, verbose=False):
  function validate_checkpoint_wrapper_import (line 24) | def validate_checkpoint_wrapper_import(checkpoint_wrapper):

FILE: build/lib/slowfast/models/video_model_builder.py
  class FuseFastToSlow (line 105) | class FuseFastToSlow(nn.Module):
    method __init__ (line 112) | def __init__(
    method forward (line 155) | def forward(self, x):
  class SlowFast (line 166) | class SlowFast(nn.Module):
    method __init__ (line 175) | def __init__(self, cfg):
    method _construct_network (line 196) | def _construct_network(self, cfg):
    method forward (line 418) | def forward(self, x, bboxes=None):
  class ResNet (line 440) | class ResNet(nn.Module):
    method __init__ (line 454) | def __init__(self, cfg):
    method _construct_network (line 475) | def _construct_network(self, cfg):
    method forward (line 638) | def forward(self, x, bboxes=None):
  class X3D (line 657) | class X3D(nn.Module):
    method __init__ (line 666) | def __init__(self, cfg):
    method _round_repeats (line 704) | def _round_repeats(self, repeats, multiplier):
    method _construct_network (line 711) | def _construct_network(self, cfg):
    method forward (line 796) | def forward(self, x, bboxes=None):
  class MViT (line 803) | class MViT(nn.Module):
    method __init__ (line 815) | def __init__(self, cfg):
    method _init_weights (line 1027) | def _init_weights(self, m):
    method no_weight_decay (line 1037) | def no_weight_decay(self):
    method _get_pos_embed (line 1060) | def _get_pos_embed(self, pos_embed, bcthw):
    method forward (line 1084) | def forward(self, x):

FILE: build/lib/slowfast/site.py
  function __boot (line 1) | def __boot():

FILE: build/lib/slowfast/utils/ava_eval_helper.py
  function make_image_key (line 49) | def make_image_key(video_id, timestamp):
  function read_csv (line 54) | def read_csv(csv_file, class_whitelist=None, load_score=False):
  function read_exclusions (line 91) | def read_exclusions(exclusions_file):
  function read_labelmap (line 109) | def read_labelmap(labelmap_file):
  function evaluate_ava_from_files (line 127) | def evaluate_ava_from_files(labelmap, groundtruth, detections, exclusions):
  function evaluate_ava (line 137) | def evaluate_ava(
  function run_evaluation (line 174) | def run_evaluation(
  function get_ava_eval_data (line 251) | def get_ava_eval_data(
  function write_results (line 290) | def write_results(detections, filename):

FILE: build/lib/slowfast/utils/ava_evaluation/label_map_util.py
  function _validate_label_map (line 29) | def _validate_label_map(label_map):
  function create_category_index (line 43) | def create_category_index(categories):
  function get_max_label_map_index (line 62) | def get_max_label_map_index(label_map):
  function convert_label_map_to_categories (line 74) | def convert_label_map_to_categories(
  function load_labelmap (line 129) | def load_labelmap(path):
  function get_label_map_dict (line 148) | def get_label_map_dict(label_map_path, use_display_name=False):
  function create_category_index_from_labelmap (line 168) | def create_category_index_from_labelmap(label_map_path):
  function create_class_agnostic_category_index (line 185) | def create_class_agnostic_category_index():

FILE: build/lib/slowfast/utils/ava_evaluation/metrics.py
  function compute_precision_recall (line 21) | def compute_precision_recall(scores, labels, num_gt):
  function compute_average_precision (line 74) | def compute_average_precision(precision, recall):
  function compute_cor_loc (line 128) | def compute_cor_loc(

FILE: build/lib/slowfast/utils/ava_evaluation/np_box_list.py
  class BoxList (line 27) | class BoxList(object):
    method __init__ (line 39) | def __init__(self, data):
    method num_boxes (line 64) | def num_boxes(self):
    method get_extra_fields (line 68) | def get_extra_fields(self):
    method has_field (line 72) | def has_field(self, field):
    method add_field (line 75) | def add_field(self, field, field_data):
    method get (line 92) | def get(self):
    method get_field (line 100) | def get_field(self, field):
    method get_coordinates (line 116) | def get_coordinates(self):
    method _is_valid_boxes (line 129) | def _is_valid_boxes(self, data):

FILE: build/lib/slowfast/utils/ava_evaluation/np_box_list_ops.py
  class SortOrder (line 33) | class SortOrder(object):
  function area (line 45) | def area(boxlist):
  function intersection (line 58) | def intersection(boxlist1, boxlist2):
  function iou (line 71) | def iou(boxlist1, boxlist2):
  function ioa (line 84) | def ioa(boxlist1, boxlist2):
  function gather (line 101) | def gather(boxlist, indices, fields=None):
  function sort_by_field (line 136) | def sort_by_field(boxlist, field, order=SortOrder.DESCEND):
  function non_max_suppression (line 167) | def non_max_suppression(
  function multi_class_non_max_suppression (line 242) | def multi_class_non_max_suppression(
  function scale (line 321) | def scale(boxlist, y_scale, x_scale):
  function clip_to_window (line 349) | def clip_to_window(boxlist, window):
  function prune_non_overlapping_boxes (line 385) | def prune_non_overlapping_boxes(boxlist1, boxlist2, minoverlap=0.0):
  function prune_outside_window (line 410) | def prune_outside_window(boxlist, window):
  function concatenate (line 448) | def concatenate(boxlists, fields=None):
  function filter_scores_greater_than (line 503) | def filter_scores_greater_than(boxlist, thresh):
  function change_coordinate_frame (line 539) | def change_coordinate_frame(boxlist, window):
  function _copy_extra_fields (line 572) | def _copy_extra_fields(boxlist_to_copy_to, boxlist_to_copy_from):
  function _update_valid_indices_by_removing_high_iou_boxes (line 589) | def _update_valid_indices_by_removing_high_iou_boxes(

FILE: build/lib/slowfast/utils/ava_evaluation/np_box_mask_list.py
  class BoxMaskList (line 29) | class BoxMaskList(np_box_list.BoxList):
    method __init__ (line 37) | def __init__(self, box_data, mask_data):
    method get_masks (line 67) | def get_masks(self):

FILE: build/lib/slowfast/utils/ava_evaluation/np_box_mask_list_ops.py
  function box_list_to_box_mask_list (line 33) | def box_list_to_box_mask_list(boxlist):
  function area (line 57) | def area(box_mask_list):
  function intersection (line 69) | def intersection(box_mask_list1, box_mask_list2):
  function iou (line 84) | def iou(box_mask_list1, box_mask_list2):
  function ioa (line 99) | def ioa(box_mask_list1, box_mask_list2):
  function gather (line 118) | def gather(box_mask_list, indices, fields=None):
  function sort_by_field (line 151) | def sort_by_field(
  function non_max_suppression (line 174) | def non_max_suppression(
  function multi_class_non_max_suppression (line 255) | def multi_class_non_max_suppression(
  function prune_non_overlapping_masks (line 337) | def prune_non_overlapping_masks(box_mask_list1, box_mask_list2, minoverl...
  function concatenate (line 365) | def concatenate(box_mask_lists, fields=None):
  function filter_scores_greater_than (line 395) | def filter_scores_greater_than(box_mask_list, thresh):

FILE: build/lib/slowfast/utils/ava_evaluation/np_box_ops.py
  function area (line 31) | def area(boxes):
  function intersection (line 43) | def intersection(boxes1, boxes2):
  function iou (line 71) | def iou(boxes1, boxes2):
  function ioa (line 92) | def ioa(boxes1, boxes2):

FILE: build/lib/slowfast/utils/ava_evaluation/np_mask_ops.py
  function area (line 33) | def area(masks):
  function intersection (line 51) | def intersection(masks1, masks2):
  function iou (line 79) | def iou(masks1, masks2):
  function ioa (line 107) | def ioa(masks1, masks2):

FILE: build/lib/slowfast/utils/ava_evaluation/object_detection_evaluation.py
  class DetectionEvaluator (line 44) | class DetectionEvaluator(object):
    method __init__ (line 64) | def __init__(self, categories):
    method add_single_ground_truth_image_info (line 75) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
    method add_single_detected_image_info (line 86) | def add_single_detected_image_info(self, image_id, detections_dict):
    method evaluate (line 97) | def evaluate(self):
    method clear (line 102) | def clear(self):
  class ObjectDetectionEvaluator (line 107) | class ObjectDetectionEvaluator(DetectionEvaluator):
    method __init__ (line 110) | def __init__(
    method add_single_ground_truth_image_info (line 158) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
    method add_single_detected_image_info (line 236) | def add_single_detected_image_info(self, image_id, detections_dict):
    method evaluate (line 287) | def evaluate(self):
    method clear (line 346) | def clear(self):
  class PascalDetectionEvaluator (line 357) | class PascalDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 360) | def __init__(self, categories, matching_iou_threshold=0.5):
  class WeightedPascalDetectionEvaluator (line 370) | class WeightedPascalDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 384) | def __init__(self, categories, matching_iou_threshold=0.5):
  class PascalInstanceSegmentationEvaluator (line 394) | class PascalInstanceSegmentationEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 397) | def __init__(self, categories, matching_iou_threshold=0.5):
  class WeightedPascalInstanceSegmentationEvaluator (line 408) | class WeightedPascalInstanceSegmentationEvaluator(ObjectDetectionEvaluat...
    method __init__ (line 422) | def __init__(self, categories, matching_iou_threshold=0.5):
  class OpenImagesDetectionEvaluator (line 433) | class OpenImagesDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 440) | def __init__(
    method add_single_ground_truth_image_info (line 460) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
  class ObjectDetectionEvaluation (line 534) | class ObjectDetectionEvaluation(object):
    method __init__ (line 537) | def __init__(
    method _initialize_detections (line 569) | def _initialize_detections(self):
    method clear_detections (line 580) | def clear_detections(self):
    method add_single_ground_truth_image_info (line 583) | def add_single_ground_truth_image_info(
    method add_single_detected_image_info (line 640) | def add_single_detected_image_info(
    method _update_ground_truth_statistics (line 727) | def _update_ground_truth_statistics(
    method evaluate (line 760) | def evaluate(self):

FILE: build/lib/slowfast/utils/ava_evaluation/per_image_evaluation.py
  class PerImageEvaluation (line 39) | class PerImageEvaluation(object):
    method __init__ (line 42) | def __init__(self, num_groundtruth_classes, matching_iou_threshold=0.5):
    method compute_object_detection_metrics (line 53) | def compute_object_detection_metrics(
    method _compute_tp_fp (line 128) | def _compute_tp_fp(
    method _get_overlaps_and_scores_box_mode (line 223) | def _get_overlaps_and_scores_box_mode(
    method _compute_tp_fp_for_single_class (line 261) | def _compute_tp_fp_for_single_class(
    method _get_ith_class_arrays (line 354) | def _get_ith_class_arrays(
    method _remove_invalid_boxes (line 411) | def _remove_invalid_boxes(

FILE: build/lib/slowfast/utils/ava_evaluation/standard_fields.py
  class InputDataFields (line 35) | class InputDataFields(object):
  class DetectionResultFields (line 100) | class DetectionResultFields(object):
  class BoxListFields (line 126) | class BoxListFields(object):
  class TfExampleFields (line 152) | class TfExampleFields(object):

FILE: build/lib/slowfast/utils/benchmark.py
  function benchmark_data_loading (line 20) | def benchmark_data_loading(cfg):

FILE: build/lib/slowfast/utils/bn_helper.py
  function compute_and_update_bn_stats (line 11) | def compute_and_update_bn_stats(model, data_loader, num_batches=200):

FILE: build/lib/slowfast/utils/c2_model_loading.py
  function get_name_convert_func (line 9) | def get_name_convert_func():

FILE: build/lib/slowfast/utils/checkpoint.py
  function make_checkpoint_dir (line 21) | def make_checkpoint_dir(path_to_job):
  function get_checkpoint_dir (line 37) | def get_checkpoint_dir(path_to_job):
  function get_path_to_checkpoint (line 46) | def get_path_to_checkpoint(path_to_job, epoch):
  function get_last_checkpoint (line 57) | def get_last_checkpoint(path_to_job):
  function has_checkpoint (line 73) | def has_checkpoint(path_to_job):
  function is_checkpoint_epoch (line 84) | def is_checkpoint_epoch(cfg, cur_epoch, multigrid_schedule=None):
  function save_checkpoint (line 107) | def save_checkpoint(path_to_job, model, optimizer, loss_scaler, epoch, c...
  function inflate_weight (line 141) | def inflate_weight(state_dict_2d, state_dict_3d):
  function load_checkpoint (line 180) | def load_checkpoint(
  function sub_to_normal_bn (line 353) | def sub_to_normal_bn(sd):
  function c2_normal_to_sub_bn (line 394) | def c2_normal_to_sub_bn(key, model_keys):
  function normal_to_sub_bn (line 414) | def normal_to_sub_bn(checkpoint_sd, model_sd):
  function load_test_checkpoint (line 455) | def load_test_checkpoint(cfg, model):
  function load_train_checkpoint (line 495) | def load_train_checkpoint(cfg, model, optimizer, loss_scaler):

FILE: build/lib/slowfast/utils/checkpoint_amp.py
  function make_checkpoint_dir (line 21) | def make_checkpoint_dir(path_to_job):
  function get_checkpoint_dir (line 37) | def get_checkpoint_dir(path_to_job):
  function get_path_to_checkpoint (line 46) | def get_path_to_checkpoint(path_to_job, epoch):
  function get_last_checkpoint (line 57) | def get_last_checkpoint(path_to_job):
  function has_checkpoint (line 73) | def has_checkpoint(path_to_job):
  function is_checkpoint_epoch (line 84) | def is_checkpoint_epoch(cfg, cur_epoch, multigrid_schedule=None):
  function save_checkpoint (line 107) | def save_checkpoint(path_to_job, model, optimizer, loss_scaler, epoch, c...
  function inflate_weight (line 141) | def inflate_weight(state_dict_2d, state_dict_3d):
  function load_checkpoint (line 180) | def load_checkpoint(
  function sub_to_normal_bn (line 353) | def sub_to_normal_bn(sd):
  function c2_normal_to_sub_bn (line 394) | def c2_normal_to_sub_bn(key, model_keys):
  function normal_to_sub_bn (line 414) | def normal_to_sub_bn(checkpoint_sd, model_sd):
  function load_test_checkpoint (line 455) | def load_test_checkpoint(cfg, model):
  function load_train_checkpoint (line 495) | def load_train_checkpoint(cfg, model, optimizer, loss_scaler):

FILE: build/lib/slowfast/utils/distributed.py
  function all_gather (line 22) | def all_gather(tensors):
  function all_reduce (line 44) | def all_reduce(tensors, average=True):
  function init_process_group (line 63) | def init_process_group(
  function is_master_proc (line 101) | def is_master_proc(num_gpus=8):
  function is_root_proc (line 111) | def is_root_proc():
  function get_rank (line 121) | def get_rank():
  function synchronize (line 132) | def synchronize():
  function _get_global_gloo_group (line 148) | def _get_global_gloo_group():
  function _serialize_to_tensor (line 161) | def _serialize_to_tensor(data, group):
  function _pad_to_largest_tensor (line 189) | def _pad_to_largest_tensor(tensor, group):
  function all_gather_unaligned (line 225) | def all_gather_unaligned(data, group=None):
  class GatherLayer (line 264) | class GatherLayer(torch.autograd.Function):
    method forward (line 268) | def forward(ctx, input):
    method backward (line 275) | def backward(ctx, *grads):
  class AllGatherWithGradient (line 282) | class AllGatherWithGradient(torch.autograd.Function):
    method forward (line 286) | def forward(ctx, input):
    method backward (line 294) | def backward(ctx, grad_output):

FILE: build/lib/slowfast/utils/env.py
  function setup_environment (line 13) | def setup_environment():

FILE: build/lib/slowfast/utils/logging.py
  function _suppress_print (line 19) | def _suppress_print():
  function _cached_log_stream (line 31) | def _cached_log_stream(filename):
  function setup_logging (line 40) | def setup_logging(output_dir=None):
  function get_logger (line 77) | def get_logger(name):
  function log_json_stats (line 87) | def log_json_stats(stats, output_dir=None):

FILE: build/lib/slowfast/utils/lr_policy.py
  function get_lr_at_epoch (line 9) | def get_lr_at_epoch(cfg, cur_epoch):
  function lr_func_cosine (line 30) | def lr_func_cosine(cfg, cur_epoch):
  function lr_func_steps_with_relative_lrs (line 56) | def lr_func_steps_with_relative_lrs(cfg, cur_epoch):
  function get_step_index (line 69) | def get_step_index(cfg, cur_epoch):
  function get_lr_func (line 84) | def get_lr_func(lr_policy):

FILE: build/lib/slowfast/utils/meters.py
  function get_ava_mini_groundtruth (line 28) | def get_ava_mini_groundtruth(full_groundtruth):
  class AVAMeter (line 46) | class AVAMeter(object):
    method __init__ (line 51) | def __init__(self, overall_iters, cfg, mode):
    method log_iter_stats (line 86) | def log_iter_stats(self, cur_epoch, cur_iter):
    method iter_tic (line 138) | def iter_tic(self):
    method iter_toc (line 145) | def iter_toc(self):
    method data_toc (line 152) | def data_toc(self):
    method reset (line 156) | def reset(self):
    method update_stats (line 166) | def update_stats(self, preds, ori_boxes, metadata, loss=None, lr=None):
    method finalize_metrics (line 185) | def finalize_metrics(self, log=True):
    method log_epoch_stats (line 212) | def log_epoch_stats(self, cur_epoch):
  class TestMeter (line 231) | class TestMeter(object):
    method __init__ (line 239) | def __init__(
    method reset (line 287) | def reset(self):
    method update_stats (line 297) | def update_stats(self, preds, labels, clip_ids):
    method log_iter_stats (line 332) | def log_iter_stats(self, cur_iter):
    method iter_tic (line 348) | def iter_tic(self):
    method iter_toc (line 355) | def iter_toc(self):
    method data_toc (line 362) | def data_toc(self):
    method finalize_metrics (line 366) | def finalize_metrics(self, ks=(1, 5)):
  class ScalarMeter (line 407) | class ScalarMeter(object):
    method __init__ (line 414) | def __init__(self, window_size):
    method reset (line 423) | def reset(self):
    method add_value (line 431) | def add_value(self, value):
    method get_win_median (line 439) | def get_win_median(self):
    method get_win_avg (line 445) | def get_win_avg(self):
    method get_global_avg (line 451) | def get_global_avg(self):
  class TrainMeter (line 458) | class TrainMeter(object):
    method __init__ (line 463) | def __init__(self, epoch_iters, cfg):
    method reset (line 487) | def reset(self):
    method iter_tic (line 500) | def iter_tic(self):
    method iter_toc (line 507) | def iter_toc(self):
    method data_toc (line 514) | def data_toc(self):
    method update_stats (line 518) | def update_stats(self, top1_err, top5_err, loss, lr, mb_size):
    method log_iter_stats (line 541) | def log_iter_stats(self, cur_epoch, cur_iter):
    method log_epoch_stats (line 571) | def log_epoch_stats(self, cur_epoch):
  class ValMeter (line 602) | class ValMeter(object):
    method __init__ (line 607) | def __init__(self, max_iter, cfg):
    method reset (line 632) | def reset(self):
    method iter_tic (line 645) | def iter_tic(self):
    method iter_toc (line 652) | def iter_toc(self):
    method data_toc (line 659) | def data_toc(self):
    method update_stats (line 663) | def update_stats(self, top1_err, top5_err, mb_size):
    method update_predictions (line 677) | def update_predictions(self, preds, labels):
    method log_iter_stats (line 688) | def log_iter_stats(self, cur_epoch, cur_iter):
    method log_epoch_stats (line 712) | def log_epoch_stats(self, cur_epoch):
  function get_map (line 744) | def get_map(preds, labels):
  class EpochTimer (line 771) | class EpochTimer:
    method __init__ (line 776) | def __init__(self) -> None:
    method reset (line 781) | def reset(self) -> None:
    method epoch_tic (line 788) | def epoch_tic(self):
    method epoch_toc (line 794) | def epoch_toc(self):
    method last_epoch_time (line 801) | def last_epoch_time(self):
    method avg_epoch_time (line 809) | def avg_epoch_time(self):
    method median_epoch_time (line 817) | def median_epoch_time(self):

FILE: build/lib/slowfast/utils/metrics.py
  function topks_correct (line 9) | def topks_correct(preds, labels, ks):
  function topk_errors (line 43) | def topk_errors(preds, labels, ks):
  function topk_accuracies (line 55) | def topk_accuracies(preds, labels, ks):

FILE: build/lib/slowfast/utils/misc.py
  function check_nan_losses (line 26) | def check_nan_losses(loss):
  function params_count (line 36) | def params_count(model, ignore_bn=False):
  function gpu_mem_usage (line 53) | def gpu_mem_usage():
  function cpu_mem_usage (line 64) | def cpu_mem_usage():
  function _get_model_analysis_input (line 78) | def _get_model_analysis_input(cfg, use_train_input):
  function get_model_stats (line 137) | def get_model_stats(model, cfg, mode, use_train_input):
  function log_model_info (line 172) | def log_model_info(model, cfg, use_train_input=True):
  function is_eval_epoch (line 200) | def is_eval_epoch(cfg, cur_epoch, multigrid_schedule):
  function plot_input (line 224) | def plot_input(tensor, bboxes=(), texts=(), path="./tmp_vis.png"):
  function frozen_bn_stats (line 254) | def frozen_bn_stats(model):
  function aggregate_sub_bn_stats (line 265) | def aggregate_sub_bn_stats(module):
  function launch_job (line 283) | def launch_job(cfg, init_method, func, daemon=False):
  function get_class_names (line 314) | def get_class_names(path, parent_path=None, subset_path=None):

FILE: build/lib/slowfast/utils/multigrid.py
  class MultigridSchedule (line 13) | class MultigridSchedule(object):
    method init_multigrid (line 18) | def init_multigrid(self, cfg):
    method update_long_cycle (line 63) | def update_long_cycle(self, cfg, cur_epoch):
    method get_long_cycle_schedule (line 123) | def get_long_cycle_schedule(self, cfg):
  function print_schedule (line 215) | def print_schedule(schedule):
  function get_current_long_cycle_shape (line 224) | def get_current_long_cycle_shape(schedule, epoch):

FILE: build/lib/slowfast/utils/multiprocessing.py
  function run (line 9) | def run(

FILE: build/lib/slowfast/utils/parser.py
  function parse_args (line 13) | def parse_args():
  function load_config (line 67) | def load_config(args):

FILE: build/lib/slowfast/utils/weight_init_helper.py
  function init_weights (line 10) | def init_weights(

FILE: build/lib/slowfast/visualization/async_predictor.py
  class AsycnActionPredictor (line 17) | class AsycnActionPredictor:
    class _Predictor (line 18) | class _Predictor(mp.Process):
      method __init__ (line 19) | def __init__(self, cfg, task_queue, result_queue, gpu_id=None):
      method run (line 41) | def run(self):
    method __init__ (line 54) | def __init__(self, cfg, result_queue=None):
    method put (line 78) | def put(self, task):
    method get (line 88) | def get(self):
    method __call__ (line 106) | def __call__(self, task):
    method shutdown (line 110) | def shutdown(self):
    method result_available (line 115) | def result_available(self):
    method default_buffer_size (line 122) | def default_buffer_size(self):
  class AsyncVis (line 126) | class AsyncVis:
    class _VisWorker (line 127) | class _VisWorker(mp.Process):
      method __init__ (line 128) | def __init__(self, video_vis, task_queue, result_queue):
      method run (line 141) | def run(self):
    method __init__ (line 154) | def __init__(self, video_vis, n_workers=None):
    method put (line 183) | def put(self, task):
    method get (line 193) | def get(self):
    method __call__ (line 213) | def __call__(self, task):
    method shutdown (line 220) | def shutdown(self):
    method result_available (line 225) | def result_available(self):
    method default_buffer_size (line 229) | def default_buffer_size(self):
  class _StopToken (line 233) | class _StopToken:
  class AsyncDemo (line 237) | class AsyncDemo:
    method __init__ (line 242) | def __init__(self, cfg, async_vis):
    method put (line 254) | def put(self, task):
    method get (line 264) | def get(self):
  function draw_predictions (line 276) | def draw_predictions(task, video_vis):

FILE: build/lib/slowfast/visualization/ava_demo_precomputed_boxes.py
  class AVAVisualizerWithPrecomputedBox (line 24) | class AVAVisualizerWithPrecomputedBox:
    method __init__ (line 30) | def __init__(self, cfg):
    method get_output_file (line 77) | def get_output_file(self, path):
    method get_input_clip (line 91) | def get_input_clip(self, keyframe_idx):
    method get_predictions (line 124) | def get_predictions(self):
    method draw_video (line 193) | def draw_video(self):
    method __call__ (line 313) | def __call__(self):
    method display (line 317) | def display(self, frame):
    method _get_keyframe_clip (line 327) | def _get_keyframe_clip(self, keyframe_idx):
    method _get_frame_range (line 339) | def _get_frame_range(self, start_idx, num_frames):
  function merge_pred_gt_boxes (line 362) | def merge_pred_gt_boxes(pred_dict, gt_dict=None):
  function load_boxes_labels (line 390) | def load_boxes_labels(cfg, video_name, fps, img_width, img_height):

FILE: build/lib/slowfast/visualization/demo_loader.py
  class VideoManager (line 17) | class VideoManager:
    method __init__ (line 22) | def __init__(self, cfg):
    method __iter__ (line 67) | def __iter__(self):
    method __next__ (line 70) | def __next__(self):
    method get_output_file (line 100) | def get_output_file(self, path, fps=30):
    method display (line 115) | def display(self, task):
    method clean (line 130) | def clean(self):
    method start (line 140) | def start(self):
    method join (line 143) | def join(self):
  class ThreadVideoManager (line 147) | class ThreadVideoManager:
    method __init__ (line 153) | def __init__(self, cfg):
    method get_output_file (line 211) | def get_output_file(self, path, fps=30):
    method __iter__ (line 226) | def __iter__(self):
    method put_fn (line 229) | def put_fn(self):
    method __next__ (line 268) | def __next__(self):
    method get_fn (line 295) | def get_fn(self):
    method display (line 325) | def display(self, task):
    method start (line 335) | def start(self):
    method join (line 350) | def join(self):
    method clean (line 353) | def clean(self):

FILE: build/lib/slowfast/visualization/gradcam_utils.py
  class GradCAM (line 12) | class GradCAM:
    method __init__ (line 19) | def __init__(
    method _register_single_hook (line 45) | def _register_single_hook(self, layer_name):
    method _register_hooks (line 63) | def _register_hooks(self):
    method _calculate_localization_map (line 70) | def _calculate_localization_map(self, inputs, labels=None):
    method __call__ (line 142) | def __call__(self, inputs, labels=None, alpha=0.5):

FILE: build/lib/slowfast/visualization/prediction_vis.py
  class WrongPredictionVis (line 16) | class WrongPredictionVis:
    method __init__ (line 22) | def __init__(self, cfg):
    method _pick_wrong_preds (line 46) | def _pick_wrong_preds(self, labels, preds):
    method visualize_vid (line 72) | def visualize_vid(self, video_input, labels, preds, batch_idx):
    method wrong_class_prediction (line 122) | def wrong_class_prediction(self):
    method clean (line 131) | def clean(self):

FILE: build/lib/slowfast/visualization/predictor.py
  class Predictor (line 20) | class Predictor:
    method __init__ (line 25) | def __init__(self, cfg, gpu_id=None):
    method __call__ (line 49) | def __call__(self, task):
  class ActionPredictor (line 119) | class ActionPredictor:
    method __init__ (line 124) | def __init__(self, cfg, async_vis=None, gpu_id=None):
    method put (line 135) | def put(self, task):
    method get (line 146) | def get(self):
  class Detectron2Predictor (line 158) | class Detectron2Predictor:
    method __init__ (line 164) | def __init__(self, cfg, gpu_id=None):
    method __call__ (line 189) | def __call__(self, task):

FILE: build/lib/slowfast/visualization/tensorboard_vis.py
  class TensorboardWriter (line 20) | class TensorboardWriter(object):
    method __init__ (line 25) | def __init__(self, cfg):
    method add_scalars (line 90) | def add_scalars(self, data_dict, global_step=None):
    method plot_eval (line 101) | def plot_eval(self, preds, labels, global_step=None):
    method add_video (line 171) | def add_video(self, vid_tensor, tag="Video Input", global_step=None, f...
    method plot_weights_and_activations (line 183) | def plot_weights_and_activations(
    method flush (line 227) | def flush(self):
    method close (line 230) | def close(self):
  function add_confusion_matrix (line 235) | def add_confusion_matrix(
  function plot_hist (line 280) | def plot_hist(
  function add_ndim_array (line 332) | def add_ndim_array(
  function add_heatmap (line 408) | def add_heatmap(tensor):

FILE: build/lib/slowfast/visualization/utils.py
  function get_confusion_matrix (line 16) | def get_confusion_matrix(preds, labels, num_classes, normalize="true"):
  function plot_confusion_matrix (line 48) | def plot_confusion_matrix(cmtx, num_classes, class_names=None, figsize=N...
  function plot_topk_histogram (line 92) | def plot_topk_histogram(tag, array, k=10, class_names=None, figsize=None):
  class GetWeightAndActivation (line 158) | class GetWeightAndActivation:
    method __init__ (line 163) | def __init__(self, model, layers):
    method _get_layer (line 178) | def _get_layer(self, layer_name):
    method _register_single_hook (line 191) | def _register_single_hook(self, layer_name):
    method _register_hooks (line 204) | def _register_hooks(self):
    method get_activations (line 211) | def get_activations(self, input, bboxes=None):
    method get_weights (line 236) | def get_weights(self):
  function get_indexing (line 255) | def get_indexing(string):
  function process_layer_index_data (line 278) | def process_layer_index_data(layer_ls, layer_name_prefix=""):
  function process_cv2_inputs (line 304) | def process_cv2_inputs(frames, cfg):
  function get_layer (line 325) | def get_layer(model, layer_name):
  class TaskInfo (line 343) | class TaskInfo:
    method __init__ (line 344) | def __init__(self):
    method add_frames (line 355) | def add_frames(self, idx, frames):
    method add_bboxes (line 365) | def add_bboxes(self, bboxes):
    method add_action_preds (line 371) | def add_action_preds(self, preds):

FILE: build/lib/slowfast/visualization/video_visualizer.py
  function _create_text_labels (line 18) | def _create_text_labels(classes, scores, class_names, ground_truth=False):
  class ImgVisualizer (line 45) | class ImgVisualizer(Visualizer):
    method __init__ (line 46) | def __init__(self, img_rgb, meta, **kwargs):
    method draw_text (line 61) | def draw_text(
    method draw_multiple_text (line 109) | def draw_multiple_text(
    method draw_multiple_text_upward (line 184) | def draw_multiple_text_upward(
    method draw_multiple_text_downward (line 237) | def draw_multiple_text_downward(
    method _align_x_coordinate (line 290) | def _align_x_coordinate(self, box_coordinate):
    method _align_y_top (line 312) | def _align_y_top(self, box_coordinate, num_text, textbox_width):
    method _align_y_bottom (line 330) | def _align_y_bottom(self, box_coordinate, num_text, textbox_width):
  class VideoVisualizer (line 349) | class VideoVisualizer:
    method __init__ (line 350) | def __init__(
    method _get_color (line 396) | def _get_color(self, class_id):
    method draw_one_frame (line 404) | def draw_one_frame(
    method draw_clip_range (line 514) | def draw_clip_range(
    method draw_clip (line 568) | def draw_clip(
    method _adjust_frames_type (line 635) | def _adjust_frames_type(self, frames):
    method _get_thres_array (line 657) | def _get_thres_array(self, common_class_names=None):

FILE: slowfast/config/custom_config.py
  function add_custom_config (line 7) | def add_custom_config(_C):

FILE: slowfast/config/defaults.py
  function assert_and_infer_cfg (line 990) | def assert_and_infer_cfg(cfg):
  function get_cfg (line 1018) | def get_cfg():

FILE: slowfast/datasets/ava_dataset.py
  class Ava (line 18) | class Ava(torch.utils.data.Dataset):
    method __init__ (line 23) | def __init__(self, cfg, split):
    method _load_data (line 49) | def _load_data(self, cfg):
    method print_summary (line 87) | def print_summary(self):
    method __len__ (line 98) | def __len__(self):
    method num_videos (line 106) | def num_videos(self):
    method _images_and_boxes_preprocessing_cv2 (line 113) | def _images_and_boxes_preprocessing_cv2(self, imgs, boxes):
    method _images_and_boxes_preprocessing (line 245) | def _images_and_boxes_preprocessing(self, imgs, boxes):
    method __getitem__ (line 352) | def __getitem__(self, idx):

FILE: slowfast/datasets/ava_helper.py
  function load_image_lists (line 16) | def load_image_lists(cfg, is_train):
  function load_boxes_and_labels (line 69) | def load_boxes_and_labels(cfg, mode):
  function get_keyframe_data (line 117) | def get_keyframe_data(boxes_and_labels):
  function get_num_boxes_used (line 163) | def get_num_boxes_used(keyframe_indices, keyframe_boxes_and_labels):
  function parse_bboxes_file (line 182) | def parse_bboxes_file(

FILE: slowfast/datasets/build.py
  function build_dataset (line 15) | def build_dataset(dataset_name, cfg, split):

FILE: slowfast/datasets/charades.py
  class Charades (line 20) | class Charades(torch.utils.data.Dataset):
    method __init__ (line 31) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 74) | def _construct_loader(self):
    method get_seq_frames (line 113) | def get_seq_frames(self, index):
    method __getitem__ (line 154) | def __getitem__(self, index):
    method __len__ (line 249) | def __len__(self):
    method num_videos (line 257) | def num_videos(self):

FILE: slowfast/datasets/cv2_transform.py
  function clip_boxes_to_image (line 9) | def clip_boxes_to_image(boxes, height, width):
  function random_short_side_scale_jitter_list (line 29) | def random_short_side_scale_jitter_list(images, min_size, max_size, boxe...
  function scale (line 77) | def scale(size, image):
  function scale_boxes (line 106) | def scale_boxes(size, boxes, height, width):
  function horizontal_flip_list (line 134) | def horizontal_flip_list(prob, images, order="CHW", boxes=None):
  function spatial_shift_crop_list (line 166) | def spatial_shift_crop_list(size, images, spatial_shift_pos, boxes=None):
  function CHW2HWC (line 216) | def CHW2HWC(image):
  function HWC2CHW (line 228) | def HWC2CHW(image):
  function color_jitter_list (line 240) | def color_jitter_list(
  function lighting_list (line 273) | def lighting_list(imgs, alphastd, eigval, eigvec, alpha=None):
  function color_normalization (line 302) | def color_normalization(image, mean, stddev):
  function pad_image (line 319) | def pad_image(image, pad_size, order="CHW"):
  function horizontal_flip (line 344) | def horizontal_flip(prob, image, order="CHW"):
  function flip_boxes (line 365) | def flip_boxes(boxes, im_width):
  function crop_boxes (line 381) | def crop_boxes(boxes, x_offset, y_offset):
  function random_crop_list (line 394) | def random_crop_list(images, size, pad_size=0, order="CHW", boxes=None):
  function center_crop (line 458) | def center_crop(size, image):
  function random_scale_jitter (line 477) | def random_scale_jitter(image, min_size, max_size):
  function random_scale_jitter_list (line 495) | def random_scale_jitter_list(images, min_size, max_size):
  function random_sized_crop (line 513) | def random_sized_crop(image, size, area_frac=0.08):
  function lighting (line 556) | def lighting(img, alphastd, eigval, eigvec):
  function random_sized_crop_list (line 582) | def random_sized_crop_list(images, size, crop_area_fraction=0.08):
  function blend (line 632) | def blend(image1, image2, alpha):
  function grayscale (line 636) | def grayscale(image):
  function saturation (line 654) | def saturation(var, image):
  function brightness (line 668) | def brightness(var, image):
  function contrast (line 682) | def contrast(var, image):
  function saturation_list (line 697) | def saturation_list(var, images):
  function brightness_list (line 715) | def brightness_list(var, images):
  function contrast_list (line 733) | def contrast_list(var, images):
  function color_jitter (line 752) | def color_jitter(image, img_brightness=0, img_contrast=0, img_saturation...
  function revert_scaled_boxes (line 783) | def revert_scaled_boxes(size, boxes, img_height, img_width):

FILE: slowfast/datasets/decoder.py
  function temporal_sampling (line 11) | def temporal_sampling(frames, start_idx, end_idx, num_samples):
  function get_start_end_idx (line 31) | def get_start_end_idx(
  function pyav_decode_stream (line 72) | def pyav_decode_stream(
  function torchvision_decode (line 114) | def torchvision_decode(
  function pyav_decode (line 229) | def pyav_decode(
  function decode (line 305) | def decode(

FILE: slowfast/datasets/imagenet.py
  class Imagenet (line 26) | class Imagenet(torch.utils.data.Dataset):
    method __init__ (line 29) | def __init__(self, cfg, mode, num_retries=10):
    method _load_imdb (line 45) | def _load_imdb(self):
    method _construct_imdb (line 53) | def _construct_imdb(self):
    method load_image (line 76) | def load_image(self, im_path):
    method _prepare_im_res (line 86) | def _prepare_im_res(self, im_path):
    method _prepare_im_tf (line 123) | def _prepare_im_tf(self, im_path):
    method __load__ (line 162) | def __load__(self, index):
    method __getitem__ (line 183) | def __getitem__(self, index):
    method __len__ (line 202) | def __len__(self):

FILE: slowfast/datasets/kinetics.py
  class Kinetics (line 24) | class Kinetics(torch.utils.data.Dataset):
    method __init__ (line 35) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 88) | def _construct_loader(self):
    method __getitem__ (line 129) | def __getitem__(self, index):
    method _aug_frame (line 320) | def _aug_frame(
    method _frame_to_list_img (line 385) | def _frame_to_list_img(self, frames):
    method _list_img_to_frames (line 391) | def _list_img_to_frames(self, img_list):
    method __len__ (line 395) | def __len__(self):
    method num_videos (line 403) | def num_videos(self):

FILE: slowfast/datasets/loader.py
  function multiple_samples_collate (line 20) | def multiple_samples_collate(batch, fold=False):
  function detection_collate (line 48) | def detection_collate(batch):
  function construct_loader (line 85) | def construct_loader(cfg, split, is_precise_bn=False):
  function shuffle_dataset (line 173) | def shuffle_dataset(loader, cur_epoch):

FILE: slowfast/datasets/mixup.py
  function convert_to_one_hot (line 22) | def convert_to_one_hot(targets, num_classes, on_value=1.0, off_value=0.0):
  function mixup_target (line 40) | def mixup_target(target, num_classes, lam=1.0, smoothing=0.0):
  function rand_bbox (line 67) | def rand_bbox(img_shape, lam, margin=0.0, count=None):
  function get_cutmix_bbox (line 90) | def get_cutmix_bbox(img_shape, lam, correct_lam=True, count=None):
  class MixUp (line 109) | class MixUp:
    method __init__ (line 117) | def __init__(
    method _get_mixup_params (line 148) | def _get_mixup_params(self):
    method _mix_batch (line 167) | def _mix_batch(self, x):
    method __call__ (line 183) | def __call__(self, x, target):

FILE: slowfast/datasets/multigrid_helper.py
  class ShortCycleBatchSampler (line 19) | class ShortCycleBatchSampler(Sampler):
    method __init__ (line 26) | def __init__(self, sampler, batch_size, drop_last, cfg):
    method __iter__ (line 68) | def __iter__(self):
    method __len__ (line 82) | def __len__(self):

FILE: slowfast/datasets/ptv_datasets.py
  class PTVDatasetWrapper (line 42) | class PTVDatasetWrapper(torch.utils.data.IterableDataset):
    method __init__ (line 47) | def __init__(self, num_videos, clips_per_video, crops_per_clip, dataset):
    method __next__ (line 61) | def __next__(self):
    method sampler (line 68) | def sampler(self):
    method __len__ (line 75) | def __len__(self):
    method num_videos (line 83) | def num_videos(self):
    method __iter__ (line 90) | def __iter__(self):
  class PackPathway (line 94) | class PackPathway(torch.nn.Module):
    method __init__ (line 100) | def __init__(self, cfg):
    method forward (line 104) | def forward(self, x: torch.Tensor):
  class DictToTuple (line 108) | class DictToTuple(torch.nn.Module):
    method __init__ (line 114) | def __init__(self, num_clips, num_crops):
    method forward (line 119) | def forward(self, x: Dict[str, torch.Tensor]):
  function div255 (line 129) | def div255(x):
  function Ptvkinetics (line 143) | def Ptvkinetics(cfg, mode):
  function process_charades_label (line 275) | def process_charades_label(x, mode, num_classes):
  function rgb2bgr (line 298) | def rgb2bgr(x):
  function Ptvcharades (line 312) | def Ptvcharades(cfg, mode):
  function Ptvssv2 (line 455) | def Ptvssv2(cfg, mode):

FILE: slowfast/datasets/rand_augment.py
  function _interpolation (line 52) | def _interpolation(kwargs):
  function _check_args_tf (line 60) | def _check_args_tf(kwargs):
  function shear_x (line 66) | def shear_x(img, factor, **kwargs):
  function shear_y (line 73) | def shear_y(img, factor, **kwargs):
  function translate_x_rel (line 80) | def translate_x_rel(img, pct, **kwargs):
  function translate_y_rel (line 88) | def translate_y_rel(img, pct, **kwargs):
  function translate_x_abs (line 96) | def translate_x_abs(img, pixels, **kwargs):
  function translate_y_abs (line 103) | def translate_y_abs(img, pixels, **kwargs):
  function rotate (line 110) | def rotate(img, degrees, **kwargs):
  function auto_contrast (line 144) | def auto_contrast(img, **__):
  function invert (line 148) | def invert(img, **__):
  function equalize (line 152) | def equalize(img, **__):
  function solarize (line 156) | def solarize(img, thresh, **__):
  function solarize_add (line 160) | def solarize_add(img, add, thresh=128, **__):
  function posterize (line 175) | def posterize(img, bits_to_keep, **__):
  function contrast (line 181) | def contrast(img, factor, **__):
  function color (line 185) | def color(img, factor, **__):
  function brightness (line 189) | def brightness(img, factor, **__):
  function sharpness (line 193) | def sharpness(img, factor, **__):
  function _randomly_negate (line 197) | def _randomly_negate(v):
  function _rotate_level_to_arg (line 202) | def _rotate_level_to_arg(level, _hparams):
  function _enhance_level_to_arg (line 209) | def _enhance_level_to_arg(level, _hparams):
  function _enhance_increasing_level_to_arg (line 214) | def _enhance_increasing_level_to_arg(level, _hparams):
  function _shear_level_to_arg (line 222) | def _shear_level_to_arg(level, _hparams):
  function _translate_abs_level_to_arg (line 229) | def _translate_abs_level_to_arg(level, hparams):
  function _translate_rel_level_to_arg (line 236) | def _translate_rel_level_to_arg(level, hparams):
  function _posterize_level_to_arg (line 244) | def _posterize_level_to_arg(level, _hparams):
  function _posterize_increasing_level_to_arg (line 251) | def _posterize_increasing_level_to_arg(level, hparams):
  function _posterize_original_level_to_arg (line 258) | def _posterize_original_level_to_arg(level, _hparams):
  function _solarize_level_to_arg (line 265) | def _solarize_level_to_arg(level, _hparams):
  function _solarize_increasing_level_to_arg (line 271) | def _solarize_increasing_level_to_arg(level, _hparams):
  function _solarize_add_level_to_arg (line 277) | def _solarize_add_level_to_arg(level, _hparams):
  class AugmentOp (line 339) | class AugmentOp:
    method __init__ (line 344) | def __init__(self, name, prob=0.5, magnitude=10, hparams=None):
    method __call__ (line 366) | def __call__(self, img_list):
  function _select_rand_weights (line 446) | def _select_rand_weights(weight_idx=0, transforms=None):
  function rand_augment_ops (line 455) | def rand_augment_ops(magnitude=10, hparams=None, transforms=None):
  class RandAugment (line 464) | class RandAugment:
    method __init__ (line 465) | def __init__(self, ops, num_layers=2, choice_weights=None):
    method __call__ (line 470) | def __call__(self, img):
  function rand_augment_transform (line 483) | def rand_augment_transform(config_str, hparams):

FILE: slowfast/datasets/random_erasing.py
  function _get_pixels (line 18) | def _get_pixels(
  class RandomErasing (line 34) | class RandomErasing:
    method __init__ (line 53) | def __init__(
    method _erase (line 87) | def _erase(self, img, chan, img_h, img_w, dtype):
    method _erase_cube (line 116) | def _erase_cube(
    method __call__ (line 158) | def __call__(self, input):

FILE: slowfast/datasets/ssv2.py
  class Ssv2 (line 22) | class Ssv2(torch.utils.data.Dataset):
    method __init__ (line 33) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 73) | def _construct_loader(self):
    method get_seq_frames (line 159) | def get_seq_frames(self, index):
    method __getitem__ (line 182) | def __getitem__(self, index):
    method __len__ (line 272) | def __len__(self):
    method num_videos (line 280) | def num_videos(self):

FILE: slowfast/datasets/sth.py
  class Sth (line 26) | class Sth(torch.utils.data.Dataset):
    method __init__ (line 37) | def __init__(self, cfg, mode, num_retries=10):
    method _construct_loader (line 86) | def _construct_loader(self):
    method get_seq_frames (line 141) | def get_seq_frames(self, index, temporal_sample_index):
    method __getitem__ (line 170) | def __getitem__(self, index):
    method _aug_frame (line 307) | def _aug_frame(
    method _frame_to_list_img (line 372) | def _frame_to_list_img(self, frames):
    method _list_img_to_frames (line 378) | def _list_img_to_frames(self, img_list):
    method __len__ (line 382) | def __len__(self):
    method num_videos (line 390) | def num_videos(self):

FILE: slowfast/datasets/transform.py
  function _pil_interp (line 33) | def _pil_interp(method):
  function random_short_side_scale_jitter (line 47) | def random_short_side_scale_jitter(
  function crop_boxes (line 104) | def crop_boxes(boxes, x_offset, y_offset):
  function random_crop (line 123) | def random_crop(images, size, boxes=None):
  function horizontal_flip (line 159) | def horizontal_flip(prob, images, boxes=None):
  function uniform_crop (line 194) | def uniform_crop(images, size, spatial_idx, boxes=None, scale_size=None):
  function clip_boxes_to_image (line 257) | def clip_boxes_to_image(boxes, height, width):
  function blend (line 279) | def blend(images1, images2, alpha):
  function grayscale (line 295) | def grayscale(images):
  function color_jitter (line 317) | def color_jitter(images, img_brightness=0, img_contrast=0, img_saturatio...
  function brightness_jitter (line 352) | def brightness_jitter(var, images):
  function contrast_jitter (line 371) | def contrast_jitter(var, images):
  function saturation_jitter (line 391) | def saturation_jitter(var, images):
  function lighting_jitter (line 410) | def lighting_jitter(images, alphastd, eigval, eigvec):
  function color_normalization (line 458) | def color_normalization(images, mean, stddev):
  function _get_param_spatial_crop (line 502) | def _get_param_spatial_crop(
  function random_resized_crop (line 544) | def random_resized_crop(
  function random_resized_crop_with_shift (line 579) | def random_resized_crop_with_shift(
  function create_random_augment (line 624) | def create_random_augment(
  function random_sized_crop_img (line 660) | def random_sized_crop_img(
  class RandomResizedCropAndInterpolation (line 695) | class RandomResizedCropAndInterpolation:
    method __init__ (line 708) | def __init__(
    method get_params (line 730) | def get_params(img, scale, ratio):
    method __call__ (line 770) | def __call__(self, img):
    method __repr__ (line 784) | def __repr__(self):
  function transforms_imagenet_train (line 802) | def transforms_imagenet_train(
  function temporal_difference (line 903) | def temporal_difference(
  function color_jitter_video_ssl (line 929) | def color_jitter_video_ssl(
  function augment_raw_frames (line 980) | def augment_raw_frames(frames, time_diff_prob=0.0, gaussian_prob=0.0):
  class GaussianBlur (line 1003) | class GaussianBlur(object):
    method __init__ (line 1006) | def __init__(self, sigma=[0.1, 2.0]):
    method __call__ (line 1009) | def __call__(self, x):
  class GaussianBlurVideo (line 1018) | class GaussianBlurVideo(object):
    method __init__ (line 1019) | def __init__(
    method __call__ (line 1025) | def __call__(self, frames):

FILE: slowfast/datasets/utils.py
  function retry_load_images (line 20) | def retry_load_images(image_paths, retry=10, backend="pytorch"):
  function get_sequence (line 51) | def get_sequence(center_idx, half_len, sample_rate, num_frames):
  function pack_pathway_output (line 74) | def pack_pathway_output(cfg, frames):
  function spatial_sampling (line 110) | def spatial_sampling(
  function as_binary_vector (line 184) | def as_binary_vector(labels, num_classes):
  function aggregate_labels (line 200) | def aggregate_labels(label_list):
  function convert_to_video_level_labels (line 215) | def convert_to_video_level_labels(labels):
  function load_image_lists (line 231) | def load_image_lists(frame_list_file, prefix="", return_list=False):
  function tensor_normalize (line 276) | def tensor_normalize(tensor, mean, std, func=None):
  function get_random_sampling_rate (line 298) | def get_random_sampling_rate(long_cycle_sampling_rate, sampling_rate):
  function revert_tensor_normalize (line 310) | def revert_tensor_normalize(tensor, mean, std):
  function create_sampler (line 327) | def create_sampler(dataset, shuffle, cfg):
  function loader_worker_init_fn (line 344) | def loader_worker_init_fn(dataset):

FILE: slowfast/datasets/video_container.py
  function get_video_container (line 9) | def get_video_container(path_to_vid, multi_thread_decode=False, backend=...

FILE: slowfast/models/attention.py
  function attention_pool (line 14) | def attention_pool(tensor, pool, thw_shape, has_cls_embed=True, norm=None):
  function get_rel_pos (line 51) | def get_rel_pos(rel_pos, d):
  function cal_rel_pos_spatial (line 67) | def cal_rel_pos_spatial(
  function cal_rel_pos_temporal (line 120) | def cal_rel_pos_temporal(attn, q, has_cls_embed, q_shape, k_shape, rel_p...
  class MultiScaleAttention (line 162) | class MultiScaleAttention(nn.Module):
    method __init__ (line 163) | def __init__(
    method forward (line 307) | def forward(self, x, thw_shape):
  class MultiScaleBlock (line 445) | class MultiScaleBlock(nn.Module):
    method __init__ (line 446) | def __init__(
    method forward (line 533) | def forward(self, x, thw_shape):

FILE: slowfast/models/batchnorm_helper.py
  function get_norm (line 16) | def get_norm(cfg):
  class SubBatchNorm3d (line 40) | class SubBatchNorm3d(nn.Module):
    method __init__ (line 51) | def __init__(self, num_splits, **args):
    method _get_aggregated_mean_std (line 72) | def _get_aggregated_mean_std(self, means, stds, n):
    method aggregate_stats (line 87) | def aggregate_stats(self):
    method forward (line 101) | def forward(self, x):

FILE: slowfast/models/build.py
  function build_model (line 18) | def build_model(cfg, gpu_id=None):

FILE: slowfast/models/common.py
  class Mlp (line 7) | class Mlp(nn.Module):
    method __init__ (line 8) | def __init__(
    method forward (line 26) | def forward(self, x):
  class Permute (line 37) | class Permute(nn.Module):
    method __init__ (line 38) | def __init__(self, dims):
    method forward (line 42) | def forward(self, x):
  function drop_path (line 46) | def drop_path(x, drop_prob: float = 0.0, training: bool = False):
  class DropPath (line 62) | class DropPath(nn.Module):
    method __init__ (line 65) | def __init__(self, drop_prob=None):
    method forward (line 69) | def forward(self, x):

FILE: slowfast/models/contrastive.py
  class ContrastiveModel (line 34) | class ContrastiveModel(nn.Module):
    method __init__ (line 39) | def __init__(self, cfg):
    method knn_mem_update (line 134) | def knn_mem_update(self, q_knn, index):
    method init_knn_labels (line 145) | def init_knn_labels(self, train_loader):
    method _update_history (line 161) | def _update_history(self):
    method _batch_shuffle (line 177) | def _batch_shuffle(self, x):
    method _batch_unshuffle (line 219) | def _batch_unshuffle(self, x, idx_restore):
    method eval_knn (line 235) | def eval_knn(self, q_knn, knn_k=200):
    method sim_loss (line 245) | def sim_loss(self, q, k):
    method momentum_anneal_cosine (line 254) | def momentum_anneal_cosine(self, epoch_exact):
    method _dequeue_and_enqueue (line 266) | def _dequeue_and_enqueue(self, keys, extra_keys=None):
    method batch_clips (line 297) | def batch_clips(self, clips):
    method compute_key_feat (line 311) | def compute_key_feat(
    method forward (line 375) | def forward(
    method _simclr_precompute_pos_neg_mask_multi (line 808) | def _simclr_precompute_pos_neg_mask_multi(self):
    method run_swav_encoder_q (line 850) | def run_swav_encoder_q(self, im):
    method get_code (line 858) | def get_code(self, out):
    method run_swav_orig_encoder_q (line 867) | def run_swav_orig_encoder_q(self, x):
    method sinkhorn (line 875) | def sinkhorn(self, Q, iters):
    method distributed_sinkhorn (line 891) | def distributed_sinkhorn(self, Q, nmb_iters):
    method KLDivLoss (line 914) | def KLDivLoss(self, out, code):
  function l2_loss (line 921) | def l2_loss(x, y):
  class Normalize (line 925) | class Normalize(nn.Module):
    method __init__ (line 926) | def __init__(self, power=2, dim=1):
    method forward (line 931) | def forward(self, x):
  class Memory (line 939) | class Memory(nn.Module):
    method __init__ (line 940) | def __init__(self, length, duration, dim, cfg):
    method resize (line 955) | def resize(self, length, duration, dim):
    method get (line 968) | def get(self, ind, time, interp=False):
    method update (line 993) | def update(self, mem, momentum, ind, time, interp=False):
    method forward (line 1040) | def forward(self, inputs):
  class Memory1D (line 1044) | class Memory1D(nn.Module):
    method __init__ (line 1045) | def __init__(self, length, duration, dim, cfg):
    method get (line 1059) | def get(self, ind, time, interp=False):
    method update (line 1071) | def update(self, mem, momentum, ind, time, interp=False):
  function cancel_swav_gradients (line 1085) | def cancel_swav_gradients(model, cfg, epoch_exact):

FILE: slowfast/models/head_helper.py
  class ResNetRoIHead (line 16) | class ResNetRoIHead(nn.Module):
    method __init__ (line 21) | def __init__(
    method forward (line 114) | def forward(self, inputs, bboxes):
  class MLPHead (line 146) | class MLPHead(nn.Module):
    method __init__ (line 147) | def __init__(
    method forward (line 188) | def forward(self, x):
  class ResNetBasicHead (line 197) | class ResNetBasicHead(nn.Module):
    method __init__ (line 206) | def __init__(
    method forward (line 309) | def forward(self, inputs):
  class X3DHead (line 357) | class X3DHead(nn.Module):
    method __init__ (line 366) | def __init__(
    method _construct_head (line 416) | def _construct_head(self, dim_in, dim_inner, dim_out, norm_module):
    method forward (line 467) | def forward(self, inputs):
  class TransformerBasicHead (line 497) | class TransformerBasicHead(nn.Module):
    method __init__ (line 502) | def __init__(
    method forward (line 556) | def forward(self, x):

FILE: slowfast/models/losses.py
  class ContrastiveLoss (line 16) | class ContrastiveLoss(nn.Module):
    method __init__ (line 17) | def __init__(self, reduction="mean"):
    method forward (line 21) | def forward(self, inputs, dummy_labels=None):
  function get_loss_func (line 38) | def get_loss_func(loss_name):

FILE: slowfast/models/morphmlp.py
  function conv_3xnxn (line 12) | def conv_3xnxn(inp, oup, kernel_size=3, stride=3):
  function conv_1xnxn (line 16) | def conv_1xnxn(inp, oup, kernel_size=3, stride=3):
  class Mlp (line 19) | class Mlp(nn.Module):
    method __init__ (line 20) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 29) | def forward(self, x):
  class MorphFC_S2 (line 38) | class MorphFC_S2(nn.Module):
    method __init__ (line 39) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 51) | def forward(self, x):
  class MorphFC_S (line 76) | class MorphFC_S(nn.Module):
    method __init__ (line 77) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 91) | def forward(self, x):
  class MorphFC_T (line 117) | class MorphFC_T(nn.Module):
    method __init__ (line 118) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 129) | def forward(self, x):
  class PermutatorBlock (line 146) | class PermutatorBlock(nn.Module):
    method __init__ (line 147) | def __init__(self, dim, segment_dim, mlp_ratio=4., qkv_bias=False, qk_...
    method forward (line 164) | def forward(self, x):
  class PatchEmbed (line 171) | class PatchEmbed(nn.Module):
    method __init__ (line 175) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=...
    method forward (line 183) | def forward(self, x):
  class Downsample (line 191) | class Downsample(nn.Module):
    method __init__ (line 195) | def __init__(self, in_embed_dim, out_embed_dim, patch_size):
    method forward (line 200) | def forward(self, x):
  class MorphMLP (line 207) | class MorphMLP(nn.Module):
    method __init__ (line 211) | def __init__(self, cfg):
    method _init_weights (line 311) | def _init_weights(self, m):
    method get_pretrained_model (line 320) | def get_pretrained_model(self, cfg):
    method forward_features (line 330) | def forward_features(self, x):
    method forward (line 350) | def forward(self, x):

FILE: slowfast/models/morphmlp_32.py
  function conv_3xnxn (line 10) | def conv_3xnxn(inp, oup, kernel_size=3, stride=3):
  function conv_1xnxn (line 14) | def conv_1xnxn(inp, oup, kernel_size=3, stride=3):
  class Mlp (line 17) | class Mlp(nn.Module):
    method __init__ (line 18) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 27) | def forward(self, x):
  class MorphFC_S2 (line 36) | class MorphFC_S2(nn.Module):
    method __init__ (line 37) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 49) | def forward(self, x):
  class MorphFC_S (line 74) | class MorphFC_S(nn.Module):
    method __init__ (line 75) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 89) | def forward(self, x):
  class MorphFC_T (line 115) | class MorphFC_T(nn.Module):
    method __init__ (line 116) | def __init__(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, ...
    method forward (line 130) | def forward(self, x):
  class PermutatorBlock (line 147) | class PermutatorBlock(nn.Module):
    method __init__ (line 148) | def __init__(self, dim, segment_dim, mlp_ratio=4., qkv_bias=False, qk_...
    method forward (line 165) | def forward(self, x):
  class PatchEmbed (line 172) | class PatchEmbed(nn.Module):
    method __init__ (line 176) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=...
    method forward (line 184) | def forward(self, x):
  class Downsample (line 192) | class Downsample(nn.Module):
    method __init__ (line 196) | def __init__(self, in_embed_dim, out_embed_dim, patch_size):
    method forward (line 201) | def forward(self, x):
  class MorphMLP_32 (line 208) | class MorphMLP_32(nn.Module):
    method __init__ (line 212) | def __init__(self, cfg):
    method _init_weights (line 309) | def _init_weights(self, m):
    method get_pretrained_model (line 318) | def get_pretrained_model(self, cfg):
    method forward_features (line 328) | def forward_features(self, x):
    method forward (line 348) | def forward(self, x):

FILE: slowfast/models/nonlocal_helper.py
  class Nonlocal (line 10) | class Nonlocal(nn.Module):
    method __init__ (line 20) | def __init__(
    method _construct_nonlocal (line 67) | def _construct_nonlocal(
    method forward (line 105) | def forward(self, x):

FILE: slowfast/models/operators.py
  class SE (line 11) | class SE(nn.Module):
    method _round_width (line 14) | def _round_width(self, width, multiplier, min_width=8, divisor=8):
    method __init__ (line 35) | def __init__(self, dim_in, ratio, relu_act=True):
    method forward (line 53) | def forward(self, x):

FILE: slowfast/models/optimizer.py
  function construct_optimizer (line 11) | def construct_optimizer(model, cfg):
  function get_epoch_lr (line 101) | def get_epoch_lr(cur_epoch, cfg):
  function set_lr (line 112) | def set_lr(optimizer, new_lr):

FILE: slowfast/models/ptv_model_builder.py
  function get_head_act (line 37) | def get_head_act(act_func):
  class PTVResNet (line 59) | class PTVResNet(nn.Module):
    method __init__ (line 64) | def __init__(self, cfg):
    method _construct_network (line 90) | def _construct_network(self, cfg):
    method forward (line 209) | def forward(self, x, bboxes=None):
  class PTVSlowFast (line 225) | class PTVSlowFast(nn.Module):
    method __init__ (line 226) | def __init__(self, cfg):
    method _construct_network (line 247) | def _construct_network(self, cfg):
    method forward (line 410) | def forward(self, x, bboxes=None):
  class PTVX3D (line 425) | class PTVX3D(nn.Module):
    method __init__ (line 426) | def __init__(self, cfg):
    method _construct_network (line 449) | def _construct_network(self, cfg):
    method forward (line 505) | def forward(self, x, bboxes=None):
  class PTVCSN (line 518) | class PTVCSN(nn.Module):
    method __init__ (line 523) | def __init__(self, cfg):
    method _construct_network (line 540) | def _construct_network(self, cfg):
    method forward (line 590) | def forward(self, x, bboxes=None):
  class PTVR2plus1D (line 603) | class PTVR2plus1D(nn.Module):
    method __init__ (line 608) | def __init__(self, cfg):
    method _construct_network (line 625) | def _construct_network(self, cfg):
    method forward (line 691) | def forward(self, x, bboxes=None):
  class PTVMViT (line 704) | class PTVMViT(nn.Module):
    method __init__ (line 709) | def __init__(self, cfg):
    method _construct_network (line 726) | def _construct_network(self, cfg):
    method forward (line 770) | def forward(self, x, bboxes=None):

FILE: slowfast/models/resnet_helper.py
  function get_trans_func (line 14) | def get_trans_func(name):
  class BasicTransform (line 29) | class BasicTransform(nn.Module):
    method __init__ (line 34) | def __init__(
    method _construct (line 77) | def _construct(self, dim_in, dim_out, stride, dilation, norm_module):
    method forward (line 110) | def forward(self, x):
  class X3DTransform (line 120) | class X3DTransform(nn.Module):
    method __init__ (line 127) | def __init__(
    method _construct (line 190) | def _construct(
    method forward (line 255) | def forward(self, x):
  class BottleneckTransform (line 261) | class BottleneckTransform(nn.Module):
    method __init__ (line 267) | def __init__(
    method _construct (line 321) | def _construct(
    method forward (line 379) | def forward(self, x):
  class ResBlock (line 397) | class ResBlock(nn.Module):
    method __init__ (line 402) | def __init__(
    method _construct (line 470) | def _construct(
    method forward (line 514) | def forward(self, x):
  class ResStage (line 526) | class ResStage(nn.Module):
    method __init__ (line 537) | def __init__(
    method _construct (line 652) | def _construct(
    method forward (line 701) | def forward(self, inputs):

FILE: slowfast/models/stem_helper.py
  function get_stem_func (line 9) | def get_stem_func(name):
  class VideoModelStem (line 20) | class VideoModelStem(nn.Module):
    method __init__ (line 26) | def __init__(
    method _construct_stem (line 98) | def _construct_stem(self, dim_in, dim_out, norm_module, stem_func_name):
    method forward (line 115) | def forward(self, x):
  class ResNetBasicStem (line 127) | class ResNetBasicStem(nn.Module):
    method __init__ (line 134) | def __init__(
    method _construct_stem (line 181) | def _construct_stem(self, dim_in, dim_out, norm_module):
    method forward (line 198) | def forward(self, x):
  class X3DStem (line 206) | class X3DStem(nn.Module):
    method __init__ (line 213) | def __init__(
    method _construct_stem (line 260) | def _construct_stem(self, dim_in, dim_out, norm_module):
    method forward (line 284) | def forward(self, x):
  class PatchEmbed (line 292) | class PatchEmbed(nn.Module):
    method __init__ (line 297) | def __init__(
    method forward (line 319) | def forward(self, x):

FILE: slowfast/models/utils.py
  function round_width (line 8) | def round_width(width, multiplier, min_width=1, divisor=1, verbose=False):
  function validate_checkpoint_wrapper_import (line 24) | def validate_checkpoint_wrapper_import(checkpoint_wrapper):

FILE: slowfast/models/video_model_builder.py
  class FuseFastToSlow (line 105) | class FuseFastToSlow(nn.Module):
    method __init__ (line 112) | def __init__(
    method forward (line 155) | def forward(self, x):
  class SlowFast (line 166) | class SlowFast(nn.Module):
    method __init__ (line 175) | def __init__(self, cfg):
    method _construct_network (line 196) | def _construct_network(self, cfg):
    method forward (line 418) | def forward(self, x, bboxes=None):
  class ResNet (line 440) | class ResNet(nn.Module):
    method __init__ (line 454) | def __init__(self, cfg):
    method _construct_network (line 475) | def _construct_network(self, cfg):
    method forward (line 638) | def forward(self, x, bboxes=None):
  class X3D (line 657) | class X3D(nn.Module):
    method __init__ (line 666) | def __init__(self, cfg):
    method _round_repeats (line 704) | def _round_repeats(self, repeats, multiplier):
    method _construct_network (line 711) | def _construct_network(self, cfg):
    method forward (line 796) | def forward(self, x, bboxes=None):
  class MViT (line 803) | class MViT(nn.Module):
    method __init__ (line 815) | def __init__(self, cfg):
    method _init_weights (line 1027) | def _init_weights(self, m):
    method no_weight_decay (line 1037) | def no_weight_decay(self):
    method _get_pos_embed (line 1060) | def _get_pos_embed(self, pos_embed, bcthw):
    method forward (line 1084) | def forward(self, x):

FILE: slowfast/utils/ava_eval_helper.py
  function make_image_key (line 49) | def make_image_key(video_id, timestamp):
  function read_csv (line 54) | def read_csv(csv_file, class_whitelist=None, load_score=False):
  function read_exclusions (line 91) | def read_exclusions(exclusions_file):
  function read_labelmap (line 109) | def read_labelmap(labelmap_file):
  function evaluate_ava_from_files (line 127) | def evaluate_ava_from_files(labelmap, groundtruth, detections, exclusions):
  function evaluate_ava (line 137) | def evaluate_ava(
  function run_evaluation (line 174) | def run_evaluation(
  function get_ava_eval_data (line 251) | def get_ava_eval_data(
  function write_results (line 290) | def write_results(detections, filename):

FILE: slowfast/utils/ava_evaluation/label_map_util.py
  function _validate_label_map (line 29) | def _validate_label_map(label_map):
  function create_category_index (line 43) | def create_category_index(categories):
  function get_max_label_map_index (line 62) | def get_max_label_map_index(label_map):
  function convert_label_map_to_categories (line 74) | def convert_label_map_to_categories(
  function load_labelmap (line 129) | def load_labelmap(path):
  function get_label_map_dict (line 148) | def get_label_map_dict(label_map_path, use_display_name=False):
  function create_category_index_from_labelmap (line 168) | def create_category_index_from_labelmap(label_map_path):
  function create_class_agnostic_category_index (line 185) | def create_class_agnostic_category_index():

FILE: slowfast/utils/ava_evaluation/metrics.py
  function compute_precision_recall (line 21) | def compute_precision_recall(scores, labels, num_gt):
  function compute_average_precision (line 74) | def compute_average_precision(precision, recall):
  function compute_cor_loc (line 128) | def compute_cor_loc(

FILE: slowfast/utils/ava_evaluation/np_box_list.py
  class BoxList (line 27) | class BoxList(object):
    method __init__ (line 39) | def __init__(self, data):
    method num_boxes (line 64) | def num_boxes(self):
    method get_extra_fields (line 68) | def get_extra_fields(self):
    method has_field (line 72) | def has_field(self, field):
    method add_field (line 75) | def add_field(self, field, field_data):
    method get (line 92) | def get(self):
    method get_field (line 100) | def get_field(self, field):
    method get_coordinates (line 116) | def get_coordinates(self):
    method _is_valid_boxes (line 129) | def _is_valid_boxes(self, data):

FILE: slowfast/utils/ava_evaluation/np_box_list_ops.py
  class SortOrder (line 33) | class SortOrder(object):
  function area (line 45) | def area(boxlist):
  function intersection (line 58) | def intersection(boxlist1, boxlist2):
  function iou (line 71) | def iou(boxlist1, boxlist2):
  function ioa (line 84) | def ioa(boxlist1, boxlist2):
  function gather (line 101) | def gather(boxlist, indices, fields=None):
  function sort_by_field (line 136) | def sort_by_field(boxlist, field, order=SortOrder.DESCEND):
  function non_max_suppression (line 167) | def non_max_suppression(
  function multi_class_non_max_suppression (line 242) | def multi_class_non_max_suppression(
  function scale (line 321) | def scale(boxlist, y_scale, x_scale):
  function clip_to_window (line 349) | def clip_to_window(boxlist, window):
  function prune_non_overlapping_boxes (line 385) | def prune_non_overlapping_boxes(boxlist1, boxlist2, minoverlap=0.0):
  function prune_outside_window (line 410) | def prune_outside_window(boxlist, window):
  function concatenate (line 448) | def concatenate(boxlists, fields=None):
  function filter_scores_greater_than (line 503) | def filter_scores_greater_than(boxlist, thresh):
  function change_coordinate_frame (line 539) | def change_coordinate_frame(boxlist, window):
  function _copy_extra_fields (line 572) | def _copy_extra_fields(boxlist_to_copy_to, boxlist_to_copy_from):
  function _update_valid_indices_by_removing_high_iou_boxes (line 589) | def _update_valid_indices_by_removing_high_iou_boxes(

FILE: slowfast/utils/ava_evaluation/np_box_mask_list.py
  class BoxMaskList (line 29) | class BoxMaskList(np_box_list.BoxList):
    method __init__ (line 37) | def __init__(self, box_data, mask_data):
    method get_masks (line 67) | def get_masks(self):

FILE: slowfast/utils/ava_evaluation/np_box_mask_list_ops.py
  function box_list_to_box_mask_list (line 33) | def box_list_to_box_mask_list(boxlist):
  function area (line 57) | def area(box_mask_list):
  function intersection (line 69) | def intersection(box_mask_list1, box_mask_list2):
  function iou (line 84) | def iou(box_mask_list1, box_mask_list2):
  function ioa (line 99) | def ioa(box_mask_list1, box_mask_list2):
  function gather (line 118) | def gather(box_mask_list, indices, fields=None):
  function sort_by_field (line 151) | def sort_by_field(
  function non_max_suppression (line 174) | def non_max_suppression(
  function multi_class_non_max_suppression (line 255) | def multi_class_non_max_suppression(
  function prune_non_overlapping_masks (line 337) | def prune_non_overlapping_masks(box_mask_list1, box_mask_list2, minoverl...
  function concatenate (line 365) | def concatenate(box_mask_lists, fields=None):
  function filter_scores_greater_than (line 395) | def filter_scores_greater_than(box_mask_list, thresh):

FILE: slowfast/utils/ava_evaluation/np_box_ops.py
  function area (line 31) | def area(boxes):
  function intersection (line 43) | def intersection(boxes1, boxes2):
  function iou (line 71) | def iou(boxes1, boxes2):
  function ioa (line 92) | def ioa(boxes1, boxes2):

FILE: slowfast/utils/ava_evaluation/np_mask_ops.py
  function area (line 33) | def area(masks):
  function intersection (line 51) | def intersection(masks1, masks2):
  function iou (line 79) | def iou(masks1, masks2):
  function ioa (line 107) | def ioa(masks1, masks2):

FILE: slowfast/utils/ava_evaluation/object_detection_evaluation.py
  class DetectionEvaluator (line 44) | class DetectionEvaluator(object):
    method __init__ (line 64) | def __init__(self, categories):
    method add_single_ground_truth_image_info (line 75) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
    method add_single_detected_image_info (line 86) | def add_single_detected_image_info(self, image_id, detections_dict):
    method evaluate (line 97) | def evaluate(self):
    method clear (line 102) | def clear(self):
  class ObjectDetectionEvaluator (line 107) | class ObjectDetectionEvaluator(DetectionEvaluator):
    method __init__ (line 110) | def __init__(
    method add_single_ground_truth_image_info (line 158) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
    method add_single_detected_image_info (line 236) | def add_single_detected_image_info(self, image_id, detections_dict):
    method evaluate (line 287) | def evaluate(self):
    method clear (line 346) | def clear(self):
  class PascalDetectionEvaluator (line 357) | class PascalDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 360) | def __init__(self, categories, matching_iou_threshold=0.5):
  class WeightedPascalDetectionEvaluator (line 370) | class WeightedPascalDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 384) | def __init__(self, categories, matching_iou_threshold=0.5):
  class PascalInstanceSegmentationEvaluator (line 394) | class PascalInstanceSegmentationEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 397) | def __init__(self, categories, matching_iou_threshold=0.5):
  class WeightedPascalInstanceSegmentationEvaluator (line 408) | class WeightedPascalInstanceSegmentationEvaluator(ObjectDetectionEvaluat...
    method __init__ (line 422) | def __init__(self, categories, matching_iou_threshold=0.5):
  class OpenImagesDetectionEvaluator (line 433) | class OpenImagesDetectionEvaluator(ObjectDetectionEvaluator):
    method __init__ (line 440) | def __init__(
    method add_single_ground_truth_image_info (line 460) | def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
  class ObjectDetectionEvaluation (line 534) | class ObjectDetectionEvaluation(object):
    method __init__ (line 537) | def __init__(
    method _initialize_detections (line 569) | def _initialize_detections(self):
    method clear_detections (line 580) | def clear_detections(self):
    method add_single_ground_truth_image_info (line 583) | def add_single_ground_truth_image_info(
    method add_single_detected_image_info (line 640) | def add_single_detected_image_info(
    method _update_ground_truth_statistics (line 727) | def _update_ground_truth_statistics(
    method evaluate (line 760) | def evaluate(self):

FILE: slowfast/utils/ava_evaluation/per_image_evaluation.py
  class PerImageEvaluation (line 39) | class PerImageEvaluation(object):
    method __init__ (line 42) | def __init__(self, num_groundtruth_classes, matching_iou_threshold=0.5):
    method compute_object_detection_metrics (line 53) | def compute_object_detection_metrics(
    method _compute_tp_fp (line 128) | def _compute_tp_fp(
    method _get_overlaps_and_scores_box_mode (line 223) | def _get_overlaps_and_scores_box_mode(
    method _compute_tp_fp_for_single_class (line 261) | def _compute_tp_fp_for_single_class(
    method _get_ith_class_arrays (line 354) | def _get_ith_class_arrays(
    method _remove_invalid_boxes (line 411) | def _remove_invalid_boxes(

FILE: slowfast/utils/ava_evaluation/standard_fields.py
  class InputDataFields (line 35) | class InputDataFields(object):
  class DetectionResultFields (line 100) | class DetectionResultFields(object):
  class BoxListFields (line 126) | class BoxListFields(object):
  class TfExampleFields (line 152) | class TfExampleFields(object):

FILE: slowfast/utils/benchmark.py
  function benchmark_data_loading (line 20) | def benchmark_data_loading(cfg):

FILE: slowfast/utils/bn_helper.py
  function compute_and_update_bn_stats (line 11) | def compute_and_update_bn_stats(model, data_loader, num_batches=200):

FILE: slowfast/utils/c2_model_loading.py
  function get_name_convert_func (line 9) | def get_name_convert_func():

FILE: slowfast/utils/checkpoint.py
  function make_checkpoint_dir (line 21) | def make_checkpoint_dir(path_to_job):
  function get_checkpoint_dir (line 37) | def get_checkpoint_dir(path_to_job):
  function get_path_to_checkpoint (line 46) | def get_path_to_checkpoint(path_to_job, epoch):
  function get_last_checkpoint (line 57) | def get_last_checkpoint(path_to_job):
  function has_checkpoint (line 73) | def has_checkpoint(path_to_job):
  function is_checkpoint_epoch (line 84) | def is_checkpoint_epoch(cfg, cur_epoch, multigrid_schedule=None):
  function save_checkpoint (line 107) | def save_checkpoint(path_to_job, model, optimizer, loss_scaler, epoch, c...
  function inflate_weight (line 141) | def inflate_weight(state_dict_2d, state_dict_3d):
  function load_checkpoint (line 180) | def load_checkpoint(
  function sub_to_normal_bn (line 353) | def sub_to_normal_bn(sd):
  function c2_normal_to_sub_bn (line 394) | def c2_normal_to_sub_bn(key, model_keys):
  function normal_to_sub_bn (line 414) | def normal_to_sub_bn(checkpoint_sd, model_sd):
  function load_test_checkpoint (line 455) | def load_test_checkpoint(cfg, model):
  function load_train_checkpoint (line 495) | def load_train_checkpoint(cfg, model, optimizer, loss_scaler):

FILE: slowfast/utils/checkpoint_amp.py
  function make_checkpoint_dir (line 21) | def make_checkpoint_dir(path_to_job):
  function get_checkpoint_dir (line 37) | def get_checkpoint_dir(path_to_job):
  function get_path_to_checkpoint (line 46) | def get_path_to_checkpoint(path_to_job, epoch):
  function get_last_checkpoint (line 57) | def get_last_checkpoint(path_to_job):
  function has_checkpoint (line 73) | def has_checkpoint(path_to_job):
  function is_checkpoint_epoch (line 84) | def is_checkpoint_epoch(cfg, cur_epoch, multigrid_schedule=None):
  function save_checkpoint (line 107) | def save_checkpoint(path_to_job, model, optimizer, loss_scaler, epoch, c...
  function inflate_weight (line 141) | def inflate_weight(state_dict_2d, state_dict_3d):
  function load_checkpoint (line 180) | def load_checkpoint(
  function sub_to_normal_bn (line 353) | def sub_to_normal_bn(sd):
  function c2_normal_to_sub_bn (line 394) | def c2_normal_to_sub_bn(key, model_keys):
  function normal_to_sub_bn (line 414) | def normal_to_sub_bn(checkpoint_sd, model_sd):
  function load_test_checkpoint (line 455) | def load_test_checkpoint(cfg, model):
  function load_train_checkpoint (line 495) | def load_train_checkpoint(cfg, model, optimizer, loss_scaler):

FILE: slowfast/utils/distributed.py
  function all_gather (line 22) | def all_gather(tensors):
  function all_reduce (line 44) | def all_reduce(tensors, average=True):
  function init_process_group (line 63) | def init_process_group(
  function is_master_proc (line 101) | def is_master_proc(num_gpus=8):
  function is_root_proc (line 111) | def is_root_proc():
  function get_rank (line 121) | def get_rank():
  function synchronize (line 132) | def synchronize():
  function _get_global_gloo_group (line 148) | def _get_global_gloo_group():
  function _serialize_to_tensor (line 161) | def _serialize_to_tensor(data, group):
  function _pad_to_largest_tensor (line 189) | def _pad_to_largest_tensor(tensor, group):
  function all_gather_unaligned (line 225) | def all_gather_unaligned(data, group=None):
  class GatherLayer (line 264) | class GatherLayer(torch.autograd.Function):
    method forward (line 268) | def forward(ctx, input):
    method backward (line 275) | def backward(ctx, *grads):
  class AllGatherWithGradient (line 282) | class AllGatherWithGradient(torch.autograd.Function):
    method forward (line 286) | def forward(ctx, input):
    method backward (line 294) | def backward(ctx, grad_output):

FILE: slowfast/utils/env.py
  function setup_environment (line 13) | def setup_environment():

FILE: slowfast/utils/logging.py
  function _suppress_print (line 19) | def _suppress_print():
  function _cached_log_stream (line 31) | def _cached_log_stream(filename):
  function setup_logging (line 40) | def setup_logging(output_dir=None):
  function get_logger (line 77) | def get_logger(name):
  function log_json_stats (line 87) | def log_json_stats(stats, output_dir=None):

FILE: slowfast/utils/lr_policy.py
  function get_lr_at_epoch (line 9) | def get_lr_at_epoch(cfg, cur_epoch):
  function lr_func_cosine (line 30) | def lr_func_cosine(cfg, cur_epoch):
  function lr_func_steps_with_relative_lrs (line 56) | def lr_func_steps_with_relative_lrs(cfg, cur_epoch):
  function get_step_index (line 69) | def get_step_index(cfg, cur_epoch):
  function get_lr_func (line 84) | def get_lr_func(lr_policy):

FILE: slowfast/utils/meters.py
  function get_ava_mini_groundtruth (line 28) | def get_ava_mini_groundtruth(full_groundtruth):
  class AVAMeter (line 46) | class AVAMeter(object):
    method __init__ (line 51) | def __init__(self, overall_iters, cfg, mode):
    method log_iter_stats (line 86) | def log_iter_stats(self, cur_epoch, cur_iter):
    method iter_tic (line 138) | def iter_tic(self):
    method iter_toc (line 145) | def iter_toc(self):
    method data_toc (line 152) | def data_toc(self):
    method reset (line 156) | def reset(self):
    method update_stats (line 166) | def update_stats(self, preds, ori_boxes, metadata, loss=None, lr=None):
    method finalize_metrics (line 185) | def finalize_metrics(self, log=True):
    method log_epoch_stats (line 212) | def log_epoch_stats(self, cur_epoch):
  class TestMeter (line 231) | class TestMeter(object):
    method __init__ (line 239) | def __init__(
    method reset (line 287) | def reset(self):
    method update_stats (line 297) | def update_stats(self, preds, labels, clip_ids):
    method log_iter_stats (line 332) | def log_iter_stats(self, cur_iter):
    method iter_tic (line 348) | def iter_tic(self):
    method iter_toc (line 355) | def iter_toc(self):
    method data_toc (line 362) | def data_toc(self):
    method finalize_metrics (line 366) | def finalize_metrics(self, ks=(1, 5)):
  class ScalarMeter (line 407) | class ScalarMeter(object):
    method __init__ (line 414) | def __init__(self, window_size):
    method reset (line 423) | def reset(self):
    method add_value (line 431) | def add_value(self, value):
    method get_win_median (line 439) | def get_win_median(self):
    method get_win_avg (line 445) | def get_win_avg(self):
    method get_global_avg (line 451) | def get_global_avg(self):
  class TrainMeter (line 458) | class TrainMeter(object):
    method __init__ (line 463) | def __init__(self, epoch_iters, cfg):
    method reset (line 487) | def reset(self):
    method iter_tic (line 500) | def iter_tic(self):
    method iter_toc (line 507) | def iter_toc(self):
    method data_toc (line 514) | def data_toc(self):
    method update_stats (line 518) | def update_stats(self, top1_err, top5_err, loss, lr, mb_size):
    method log_iter_stats (line 541) | def log_iter_stats(self, cur_epoch, cur_iter):
    method log_epoch_stats (line 571) | def log_epoch_stats(self, cur_epoch):
  class ValMeter (line 602) | class ValMeter(object):
    method __init__ (line 607) | def __init__(self, max_iter, cfg):
    method reset (line 632) | def reset(self):
    method iter_tic (line 645) | def iter_tic(self):
    method iter_toc (line 652) | def iter_toc(self):
    method data_toc (line 659) | def data_toc(self):
    method update_stats (line 663) | def update_stats(self, top1_err, top5_err, mb_size):
    method update_predictions (line 677) | def update_predictions(self, preds, labels):
    method log_iter_stats (line 688) | def log_iter_stats(self, cur_epoch, cur_iter):
    method log_epoch_stats (line 712) | def log_epoch_stats(self, cur_epoch):
  function get_map (line 744) | def get_map(preds, labels):
  class EpochTimer (line 771) | class EpochTimer:
    method __init__ (line 776) | def __init__(self) -> None:
    method reset (line 781) | def reset(self) -> None:
    method epoch_tic (line 788) | def epoch_tic(self):
    method epoch_toc (line 794) | def epoch_toc(self):
    method last_epoch_time (line 801) | def last_epoch_time(self):
    method avg_epoch_time (line 809) | def avg_epoch_time(self):
    method median_epoch_time (line 817) | def median_epoch_time(self):

FILE: slowfast/utils/metrics.py
  function topks_correct (line 9) | def topks_correct(preds, labels, ks):
  function topk_errors (line 43) | def topk_errors(preds, labels, ks):
  function topk_accuracies (line 55) | def topk_accuracies(preds, labels, ks):

FILE: slowfast/utils/misc.py
  function check_nan_losses (line 26) | def check_nan_losses(loss):
  function params_count (line 36) | def params_count(model, ignore_bn=False):
  function gpu_mem_usage (line 53) | def gpu_mem_usage():
  function cpu_mem_usage (line 64) | def cpu_mem_usage():
  function _get_model_analysis_input (line 78) | def _get_model_analysis_input(cfg, use_train_input):
  function get_model_stats (line 137) | def get_model_stats(model, cfg, mode, use_train_input):
  function log_model_info (line 172) | def log_model_info(model, cfg, use_train_input=True):
  function is_eval_epoch (line 200) | def is_eval_epoch(cfg, cur_epoch, multigrid_schedule):
  function plot_input (line 224) | def plot_input(tensor, bboxes=(), texts=(), path="./tmp_vis.png"):
  function frozen_bn_stats (line 254) | def frozen_bn_stats(model):
  function aggregate_sub_bn_stats (line 265) | def aggregate_sub_bn_stats(module):
  function launch_job (line 283) | def launch_job(cfg, init_method, func, daemon=False):
  function get_class_names (line 314) | def get_class_names(path, parent_path=None, subset_path=None):

FILE: slowfast/utils/multigrid.py
  class MultigridSchedule (line 13) | class MultigridSchedule(object):
    method init_multigrid (line 18) | def init_multigrid(self, cfg):
    method update_long_cycle (line 63) | def update_long_cycle(self, cfg, cur_epoch):
    method get_long_cycle_schedule (line 123) | def get_long_cycle_schedule(self, cfg):
  function print_schedule (line 215) | def print_schedule(schedule):
  function get_current_long_cycle_shape (line 224) | def get_current_long_cycle_shape(schedule, epoch):

FILE: slowfast/utils/multiprocessing.py
  function run (line 9) | def run(

FILE: slowfast/utils/parser.py
  function parse_args (line 13) | def parse_args():
  function load_config (line 67) | def load_config(args):

FILE: slowfast/utils/weight_init_helper.py
  function init_weights (line 10) | def init_weights(

FILE: slowfast/visualization/async_predictor.py
  class AsycnActionPredictor (line 17) | class AsycnActionPredictor:
    class _Predictor (line 18) | class _Predictor(mp.Process):
      method __init__ (line 19) | def __init__(self, cfg, task_queue, result_queue, gpu_id=None):
      method run (line 41) | def run(self):
    method __init__ (line 54) | def __init__(self, cfg, result_queue=None):
    method put (line 78) | def put(self, task):
    method get (line 88) | def get(self):
    method __call__ (line 106) | def __call__(self, task):
    method shutdown (line 110) | def shutdown(self):
    method result_available (line 115) | def result_available(self):
    method default_buffer_size (line 122) | def default_buffer_size(self):
  class AsyncVis (line 126) | class AsyncVis:
    class _VisWorker (line 127) | class _VisWorker(mp.Process):
      method __init__ (line 128) | def __init__(self, video_vis, task_queue, result_queue):
      method run (line 141) | def run(self):
    method __init__ (line 154) | def __init__(self, video_vis, n_workers=None):
    method put (line 183) | def put(self, task):
    method get (line 193) | def get(self):
    method __call__ (line 213) | def __call__(self, task):
    method shutdown (line 220) | def shutdown(self):
    method result_available (line 225) | def result_available(self):
    method default_buffer_size (line 229) | def default_buffer_size(self):
  class _StopToken (line 233) | class _StopToken:
  class AsyncDemo (line 237) | class AsyncDemo:
    method __init__ (line 242) | def __init__(self, cfg, async_vis):
    method put (line 254) | def put(self, task):
    method get (line 264) | def get(self):
  function draw_predictions (line 276) | def draw_predictions(task, video_vis):

FILE: slowfast/visualization/ava_demo_precomputed_boxes.py
  class AVAVisualizerWithPrecomputedBox (line 24) | class AVAVisualizerWithPrecomputedBox:
    method __init__ (line 30) | def __init__(self, cfg):
    method get_output_file (line 77) | def get_output_file(self, path):
    method get_input_clip (line 91) | def get_input_clip(self, keyframe_idx):
    method get_predictions (line 124) | def get_predictions(self):
    method draw_video (line 193) | def draw_video(self):
    method __call__ (line 313) | def __call__(self):
    method display (line 317) | def display(self, frame):
    method _get_keyframe_clip (line 327) | def _get_keyframe_clip(self, keyframe_idx):
    method _get_frame_range (line 339) | def _get_frame_range(self, start_idx, num_frames):
  function merge_pred_gt_boxes (line 362) | def merge_pred_gt_boxes(pred_dict, gt_dict=None):
  function load_boxes_labels (line 390) | def load_boxes_labels(cfg, video_name, fps, img_width, img_height):

FILE: slowfast/visualization/demo_loader.py
  class VideoManager (line 17) | class VideoManager:
    method __init__ (line 22) | def __init__(self, cfg):
    method __iter__ (line 67) | def __iter__(self):
    method __next__ (line 70) | def __next__(self):
    method get_output_file (line 100) | def get_output_file(self, path, fps=30):
    method display (line 115) | def display(self, task):
    method clean (line 130) | def clean(self):
    method start (line 140) | def start(self):
    method join (line 143) | def join(self):
  class ThreadVideoManager (line 147) | class ThreadVideoManager:
    method __init__ (line 153) | def __init__(self, cfg):
    method get_output_file (line 211) | def get_output_file(self, path, fps=30):
    method __iter__ (line 226) | def __iter__(self):
    method put_fn (line 229) | def put_fn(self):
    method __next__ (line 268) | def __next__(self):
    method get_fn (line 295) | def get_fn(self):
    method display (line 325) | def display(self, task):
    method start (line 335) | def start(self):
    method join (line 350) | def join(self):
    method clean (line 353) | def clean(self):

FILE: slowfast/visualization/gradcam_utils.py
  class GradCAM (line 12) | class GradCAM:
    method __init__ (line 19) | def __init__(
    method _register_single_hook (line 45) | def _register_single_hook(self, layer_name):
    method _register_hooks (line 63) | def _register_hooks(self):
    method _calculate_localization_map (line 70) | def _calculate_localization_map(self, inputs, labels=None):
    method __call__ (line 142) | def __call__(self, inputs, labels=None, alpha=0.5):

FILE: slowfast/visualization/prediction_vis.py
  class WrongPredictionVis (line 16) | class WrongPredictionVis:
    method __init__ (line 22) | def __init__(self, cfg):
    method _pick_wrong_preds (line 46) | def _pick_wrong_preds(self, labels, preds):
    method visualize_vid (line 72) | def visualize_vid(self, video_input, labels, preds, batch_idx):
    method wrong_class_prediction (line 122) | def wrong_class_prediction(self):
    method clean (line 131) | def clean(self):

FILE: slowfast/visualization/predictor.py
  class Predictor (line 20) | class Predictor:
    method __init__ (line 25) | def __init__(self, cfg, gpu_id=None):
    method __call__ (line 49) | def __call__(self, task):
  class ActionPredictor (line 119) | class ActionPredictor:
    method __init__ (line 124) | def __init__(self, cfg, async_vis=None, gpu_id=None):
    method put (line 135) | def put(self, task):
    method get (line 146) | def get(self):
  class Detectron2Predictor (line 158) | class Detectron2Predictor:
    method __init__ (line 164) | def __init__(self, cfg, gpu_id=None):
    method __call__ (line 189) | def __call__(self, task):

FILE: slowfast/visualization/tensorboard_vis.py
  class TensorboardWriter (line 20) | class TensorboardWriter(object):
    method __init__ (line 25) | def __init__(self, cfg):
    method add_scalars (line 90) | def add_scalars(self, data_dict, global_step=None):
    method plot_eval (line 101) | def plot_eval(self, preds, labels, global_step=None):
    method add_video (line 171) | def add_video(self, vid_tensor, tag="Video Input", global_step=None, f...
    method plot_weights_and_activations (line 183) | def plot_weights_and_activations(
    method flush (line 227) | def flush(self):
    method close (line 230) | def close(self):
  function add_confusion_matrix (line 235) | def add_confusion_matrix(
  function plot_hist (line 280) | def plot_hist(
  function add_ndim_array (line 332) | def add_ndim_array(
  function add_heatmap (line 408) | def add_heatmap(tensor):

FILE: slowfast/visualization/utils.py
  function get_confusion_matrix (line 16) | def get_confusion_matrix(preds, labels, num_classes, normalize="true"):
  function plot_confusion_matrix (line 48) | def plot_confusion_matrix(cmtx, num_classes, class_names=None, figsize=N...
  function plot_topk_histogram (line 92) | def plot_topk_histogram(tag, array, k=10, class_names=None, figsize=None):
  class GetWeightAndActivation (line 158) | class GetWeightAndActivation:
    method __init__ (line 163) | def __init__(self, model, layers):
    method _get_layer (line 178) | def _get_layer(self, layer_name):
    method _register_single_hook (line 191) | def _register_single_hook(self, layer_name):
    method _register_hooks (line 204) | def _register_hooks(self):
    method get_activations (line 211) | def get_activations(self, input, bboxes=None):
    method get_weights (line 236) | def get_weights(self):
  function get_indexing (line 255) | def get_indexing(string):
  function process_layer_index_data (line 278) | def process_layer_index_data(layer_ls, layer_name_prefix=""):
  function process_cv2_inputs (line 304) | def process_cv2_inputs(frames, cfg):
  function get_layer (line 325) | def get_layer(model, layer_name):
  class TaskInfo (line 343) | class TaskInfo:
    method __init__ (line 344) | def __init__(self):
    method add_frames (line 355) | def add_frames(self, idx, frames):
    method add_bboxes (line 365) | def add_bboxes(self, bboxes):
    method add_action_preds (line 371) | def add_action_preds(self, preds):

FILE: slowfast/visualization/video_visualizer.py
  function _create_text_labels (line 18) | def _create_text_labels(classes, scores, class_names, ground_truth=False):
  class ImgVisualizer (line 45) | class ImgVisualizer(Visualizer):
    method __init__ (line 46) | def __init__(self, img_rgb, meta, **kwargs):
    method draw_text (line 61) | def draw_text(
    method draw_multiple_text (line 109) | def draw_multiple_text(
    method draw_multiple_text_upward (line 184) | def draw_multiple_text_upward(
    method draw_multiple_text_downward (line 237) | def draw_multiple_text_downward(
    method _align_x_coordinate (line 290) | def _align_x_coordinate(self, box_coordinate):
    method _align_y_top (line 312) | def _align_y_top(self, box_coordinate, num_text, textbox_width):
    method _align_y_bottom (line 330) | def _align_y_bottom(self, box_coordinate, num_text, textbox_width):
  class VideoVisualizer (line 349) | class VideoVisualizer:
    method __init__ (line 350) | def __init__(
    method _get_color (line 396) | def _get_color(self, class_id):
    method draw_one_frame (line 404) | def draw_one_frame(
    method draw_clip_range (line 514) | def draw_clip_range(
    method draw_clip (line 568) | def draw_clip(
    method _adjust_frames_type (line 635) | def _adjust_frames_type(self, frames):
    method _get_thres_array (line 657) | def _get_thres_array(self, common_class_names=None):

FILE: tools/benchmark.py
  function main (line 15) | def main():

FILE: tools/demo_net.py
  function run_demo (line 21) | def run_demo(cfg, frame_provider):
  function demo (line 96) | def demo(cfg):

FILE: tools/run_net.py
  function main (line 13) | def main():

FILE: tools/test_net.py
  function perform_test (line 25) | def perform_test(test_loader, model, test_meter, cfg, writer=None):
  function test (line 140) | def test(cfg):

FILE: tools/train_net.py
  function train_epoch (line 28) | def train_epoch(
  function eval_epoch (line 189) | def eval_epoch(val_loader, model, val_meter, loss_scaler, cur_epoch, cfg...
  function calculate_and_update_precise_bn (line 310) | def calculate_and_update_precise_bn(loader, model, num_iters=200, use_gp...
  function build_trainer (line 334) | def build_trainer(cfg):
  function train (line 384) | def train(cfg):

FILE: tools/visualization.py
  function run_visualization (line 29) | def run_visualization(vis_loader, model, cfg, writer=None):
  function perform_wrong_prediction_vis (line 190) | def perform_wrong_prediction_vis(vis_loader, model, cfg):
  function visualize (line 250) | def visualize(cfg):