basicsr.data.paired_image_dataset
- class basicsr.data.paired_image_dataset.PairedImageDataset(opt)[source]
Bases:
Dataset
Paired image dataset for image restoration.
Read LQ (Low Quality, e.g. LR (Low Resolution), blurry, noisy, etc) and GT image pairs.
There are three modes: 1. ‘lmdb’: Use lmdb files.
If opt[‘io_backend’] == lmdb.
- ‘meta_info_file’: Use meta information file to generate paths.
If opt[‘io_backend’] != lmdb and opt[‘meta_info_file’] is not None.
- ‘folder’: Scan folders to generate paths.
The rest.
- Parameters
opt (dict) –
Config for train datasets. It contains the following keys: dataroot_gt (str): Data root path for gt. dataroot_lq (str): Data root path for lq. meta_info_file (str): Path for meta information file. io_backend (dict): IO backend type and other kwarg. filename_tmpl (str): Template for each filename. Note that the template excludes the file extension.
Default: ‘{}’.
gt_size (int): Cropped patched size for gt patches. use_hflip (bool): Use horizontal flips. use_rot (bool): Use rotation (use vertical flip and transposing h and w for implementation).
scale (bool): Scale, which will be added automatically. phase (str): ‘train’ or ‘val’.