Calculate SSIM (structural similarity) for one channel images. It is called by func:`calculate_ssim`. Args: img1 (ndarray): Images w

Augment: horizontal flips OR rotate (0, 90, 180, 270 degrees). We use vertical flip and transpose for rotation implementation. All the images

Bundles submission data for raw denoising submission_folder Folder where denoised images reside Output is written to <submission_folder

Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_f

Bundles submission data for sRGB denoising submission_folder Folder where denoised images reside Output is written to <submission_f

Calculate NIQE (Natural Image Quality Evaluator) metric. Ref: Making a "Completely Blind" Image Quality Analyzer. This implementation could p

Calculate PSNR (Peak Signal-to-Noise Ratio). Ref: https://en.wikipedia.org/wiki/Peak_signal-to-noise_ratio Args: img1 (ndarray/tenso

calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255]

Calculate SSIM (structural similarity). Ref: Image quality assessment: From error visibility to structural similarity The results are th

get the image data and update pbar.

Crop borders of images. Args: imgs (list[ndarray] | ndarray): Images with shape (h, w, c). crop_border (int): Crop border for eac

Args: img_path (str): Image path. upsample_num_times (int): Upsamples the image before running the fa

Download files from google drive. Ref: https://stackoverflow.com/questions/25010369/wget-curl-large-file-from-google-drive # noqa E501

Downsamping with Gaussian kernel used in the DUF official code. Args: x (Tensor): Frames to be downsampled, with shape (b, t, c, h, w).

Warp an image or feature map with optical flow. Args: x (Tensor): Tensor with size (n, c, h, w). flow (Tensor): Tensor with size

Read an optical flow map. Args: flow_path (ndarray or str): Flow path. quantize (bool): whether to read quantized pair, if set to

Write optical flow to file. If the flow is not quantized, it will be saved as a .flo file losslessly, otherwise a jpeg image which is lossy b

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.