Return visualized frames/clips in the correct order based on task id if result(s) is available. Otherwise, raise queue.Empty exceptio

Reset the Meter.

Reset the Meter.

Convenience function for accesssing box coordinates. Returns: a numpy array of shape [N, 4] representing box corners

Accesses data associated with the specified field in the box collection. Args: field: a string parameter used to speficy a related

Return number of boxes held in collections.

Add data to a specified field. Args: field: a string parameter used to speficy a related field to be accessed. field_data

Convenience function for accessing masks. Returns: a numpy array of shape [N, height, width] representing masks

Add the new task to task queue. Args: task (TaskInfo object): task object that contain the necessary info

Args: cfg (CfgNode): model building configs, details are in the comments of the config file. Returns: nn.Module:

Gather boxes from BoxList according to indices and return new BoxList. By default, gather returns boxes corresponding to the input index list, as

Run 'func' on one or more GPUs, specified in cfg Args: cfg (CfgNode): configs. Details can be found in slowfast/config/de

Load the checkpoint from the given file. If inflation is True, inflate the 2D Conv weights from the checkpoint to 3D Conv. Args:

Load the checkpoint from the given file. If inflation is True, inflate the 2D Conv weights from the checkpoint to 3D Conv. Args:

Load the checkpoint from the given file. If inflation is True, inflate the 2D Conv weights from the checkpoint to 3D Conv. Args:

Constructor. Args: categories: A list of dicts, each of which has the following keys - 'id': (required) an integer id u

Add a new scalar value to the deque.

Add a new scalar value to the deque.

Builds the video model. Args: cfg (configs): configs that contains the hyper-parameters to build the backbone. Details can be

Calculate the current median value of the deque.

Calculate the current median value of the deque.

Start to record time.

Load the checkpoint from the given file. If inflation is True, inflate the 2D Conv weights from the checkpoint to 3D Conv. Args:

Update the current stats. Args: preds (tensor): prediction embedding. ori_boxes (tensor): original boxes (x1,

With 50% prob, negate the value

Get video randaug transform. Args: input_size: The size of the input video in tuple. auto_augment: Parameters for randaug. A

Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.

Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.

Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.

Return all non-box fields.

Return the actual head activation function given the activation fucntion name. Args: act_func (string): activation function to use.

Stop to record time.

Scale the short side of the image to size. Args: size (int): size to scale the image. image (array): image to perform short s

The `__init__` method of any subclass should also contain these arguments. Args: cfg (CfgNode): model buildi

The `__init__` method of any subclass should also contain these arguments. Args: cfg (CfgNode): model buildi

ResBlock class constructs redisual blocks. More details can be found in: Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun.

Given scale, ratio, height and width, return sampled coordinates of the videos.

Add multiple scalars to Tensorboard logs. Args: data_dict (dict): key is a string specifying the tag of value.

Converts a BoxList containing 'masks' into a BoxMaskList. Args: boxlist: An np_box_list.BoxList object. Returns: An np_box_mask_

Gather boxes from np_box_mask_list.BoxMaskList according to indices. By default, gather returns boxes corresponding to the input index list, as

Get path for storing checkpoints. Args: path_to_job (string): the path to the folder of the current job.

Get the function to convert Caffe2 layer names to PyTorch layer names. Returns: (func): function to convert parameter name from Caffe

Sample a clip of size clip_size from a video of size video_size and return the indices of the first and last frame of the clip. If clip_idx i

Convert the image to gray scale. Args: image (tensor): image to convert to gray scale. Dimension is `channel` x `height`

Given the arguemnts, load and initialize the configs. Args: args (argument): arguments includes `shard_id`, `num_shards`,

Log the stats. Args: cur_epoch (int): the current epoch. cur_iter (int): the current iteration.

Loads boxes and class labels from a CSV file in the AVA format. CSV file format described at https://research.google.com/ava/download.html. Ar

Convert the Sub-BN paprameters to normal BN parameters in a state dict. There are two copies of BN layers in a Sub-BN implementation: `bn.bn`

The `__init__` method of any subclass should also contain these arguments. Args: dim_in (int): the channel dimension of

Calculate and plot confusion matrix to a SummaryWriter. Args: writer (SummaryWriter): the SummaryWriter to write the matrix to.

Add heatmap to 2D tensor. Args: tensor (tensor): a 2D tensor. Tensor value must be in [0..1] range. Returns: heatmap (ten

Adds groundtruth for a single image to be used for evaluation. Args: image_id: A unique string/integer identifier for the image.

Computes area of boxes. Args: boxes: Numpy array with shape [N, 4] holding N boxes Returns: a numpy array with shape [N*1] repre

Computes area of masks. Args: masks: Numpy array with shape [N, height, width] holding N masks. Masks values are of type np.uint8 a

Blend two images with a given weight alpha. Args: images1 (tensor): the first images to be blended, the dimension is `num

Clears the state to prepare for a fresh evaluation.

Return the targeted layer (nn.Module Object) given a hierarchical layer name, separated by /. Args: model (model): model to get l

Parse the following arguments for a default parser for PySlowFast users. Args: shard_id (int): shard id for the current machine. Star

Parse AVA bounding boxes files. Args: ann_filenames (list of str(s)): a list of AVA bounding boxes annotation files. ann_is_g

Plot confusion matrices and histograms for eval/test set. Args: preds (tensor or list of tensors): list of predictions.

Reads a CSV file of excluded timestamps. Args: exclusions_file: A file object containing a csv of video-id,timestamp. Returns: A s

Read label map and class ids.

Args: cfg (CfgNode): configs. Details can be found in slowfast/config/defaults.py n_workers (Optional

The `__init__` method of any subclass should also contain these arguments. X3DHead takes a 5-dim feature tensor (BxCxTxHx

Copies the extra fields of boxlist_to_copy_from to boxlist_to_copy_to. Args: boxlist_to_copy_to: BoxList to which extra fields are copied.