Mod crop images, used during testing. Args: img (ndarray): Input image. scale (int): Scale factor. Returns: ndarray:

Model to device. It also warps models with DistributedDataParallel or DataParallel. Args: net (nn.Module)

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

Support OrderedDict for yaml. Returns: yaml Loader and Dumper.

Generate paired paths from folders. Args: folders (list[str]): A list of folder path. The order of list should be [input_fold

Generate paired paths from folders. Args: folders (list[str]): A list of folder path. The order of list should be [input_fold

Generate paired paths from lmdb files. Contents of lmdb. Taking the `lq.lmdb` for example, the file structure is: lq.lmdb ├── data.mdb

Generate paired paths from an meta information file. Each line in the meta information file contains the image names and image shape (usually

Parse option file. Args: opt_path (str): Option file path. is_train (str): Indicate whether in training or not. Default: True.

Print the str and parameter number of a network. Args: net (nn.Module)

Quantize an array of (-inf, inf) to [0, levels-1]. Args: arr (ndarray): Input array. min_val (scalar): Minimum value to be clippe

Quantize flow to [0, 255]. After this step, the size of flow will be much smaller, and can be dumped as jpeg images. Args: flow

Read image worker. Args: path (str): Image path. key (str): Image key. compress_level (int): Compress level when encoding

Reduce loss as specified. Args: loss (Tensor): Elementwise loss tensor. reduction (str): Options are 'none', 'mean' and 'sum'.

reduce loss dict. In distributed training, it averages the losses among different GPUs . Args: loss_dict (OrderedDict):

Reload the optimizers and schedulers for resumed training. Args: resume_state (dict): Resume state.

Save training states during training, which will be used for resuming. Args: epoch (int): Current epoch. curr

Set random seeds.

Set up schedulers.

Update learning rate. Args: current_iter (int): Current iteration. warmup_iter (int)： Warmup iter numbers. -1 for no

Apply element-wise weight and reduce loss. Args: loss (Tensor): Element-wise loss. weight (Tensor): Element-wise weights. Default

Format logging message. Args: log_vars (dict): It contains the following keys: epoch (int): Epoch number.