pyiqa.models.inference_model¶

Module Contents¶

class pyiqa.models.inference_model.InferenceModel(metric_name, as_loss=False, loss_weight=None, loss_reduction='mean', device=None, seed=123, check_input_range=True, **kwargs)[source]¶

Bases: torch.nn.Module

Common interface for quality inference of images with default setting of each metric.

load_weights(weights_path, weight_keys='params')[source]¶

is_valid_input(x)[source]¶

forward(target, ref=None, **kwargs)[source]¶