Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te

Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te

Generate an index list for reading `num_frames` frames from a sequence of images. Args: crt_idx (int): Current center index.

Get values according to the filepath from one lmdb named client_key. Args: filepath (str | obj:`Path`): Here, filepath is the lmd

Get 68 densemarks for cropped images. Should only have one face at most in the cropped image.

Rotate image. Args: img (ndarray): Image to be rotated. angle (float): Rotation angle in degrees. Positive values mean

imresize function same as MATLAB. It now only supports bicubic. The same scale applies for both height and width. Args: img (Ten

Initialize the dlib detectors and predictors.

Make layers by stacking the same blocks. Args: basic_block (nn.module): nn.module class for basic block. num_basic_block (int): n

Paired random crop. It crops lists of lq and gt images with corresponding locations. Args: img_gts (list[ndarray] | ndarray): GT ima

Generate paths from folder. Args: folder (str): Folder path. Returns: list[str]: Returned path list.

Generate paths from lmdb. Args: folder (str): Folder path. Returns: list[str]: Returned path list.

Pixel unshuffle. Args: x (Tensor): Input feature with shape (b, c, hh, hw). scale (int): Downsample ratio. Returns:

Read a sequence of images from a given folder path. Args: path (list[str] | str): List of image paths or image folder path. requi

Resize a flow according to ratio or shape. Args: flow (Tensor): Precomputed flow. shape [N, 2, H, W]. size_type (str): 'ratio' or

Convert a RGB image to YCbCr image. This function produces the same results as Matlab's `rgb2ycbcr` function. It implements the ITU-R BT.601

Scan a directory to find the interested files. Args: dir_path (str): Path of the directory. keywords (str | tuple(str), optional)

Get affine matrix, warp and cropped faces. Also get inverse affine matrix for post-processing.

Create a weighted version of a given loss function. To use this decorator, the loss function must have the signature like `loss_func(pred, ta

Worker for each process. Args: path (str): Image path. opt (dict): Configuration dict. It contains: crop_size (int):

Worker for each process. Args: path (str): Image path. opt (dict): Configuration dict. It contains: crop_size (int):

Convert a YCbCr image to BGR image. The bgr version of ycbcr2rgb. It implements the ITU-R BT.601 conversion for standard-definition telev

Convert a YCbCr image to RGB image. This function produces the same results as Matlab's ycbcr2rgb function. It implements the ITU-R BT.601 co