pyiqa.data.base_iqa_dataset

Module Contents

class pyiqa.data.base_iqa_dataset.BaseIQADataset(opt)[source]

Bases: torch.utils.data.Dataset

General No Reference dataset with meta info file.

Parameters:

opt (dict) – Config for train datasets with the following keys: phase (str): ‘train’ or ‘val’.

init_path_mos(opt)[source]
get_split_with_file(opt)[source]
get_split(opt)[source]

Read train/val/test splits

mos_normalize(opt)[source]
get_transforms(opt)[source]