basicsr.data.reds_dataset

class basicsr.data.reds_dataset.REDSDataset(opt)[source]

Bases: Dataset

REDS dataset for training.

The keys are generated from a meta info txt file. basicsr/data/meta_info/meta_info_REDS_GT.txt

Each line contains: 1. subfolder (clip) name; 2. frame number; 3. image shape, separated by a white space. Examples: 000 100 (720,1280,3) 001 100 (720,1280,3) …

Key examples: “000/00000000” GT (gt): Ground-Truth; LQ (lq): Low-Quality, e.g., low-resolution/blurry/noisy/compressed frames.

Parameters:

  • opt (dict) – Config for train dataset. It contains the following keys:

  • dataroot_gt (str) – Data root path for gt.

  • dataroot_lq (str) – Data root path for lq.

  • dataroot_flow (str, optional) – Data root path for flow.

  • meta_info_file (str) – Path for meta information file.

  • val_partition (str) – Validation partition types. ‘REDS4’ or ‘official’.

  • io_backend (dict) – IO backend type and other kwarg.

  • num_frame (int) – Window size for input frames.

  • gt_size (int) – Cropped patched size for gt patches.

  • interval_list (list) – Interval list for temporal augmentation.

  • random_reverse (bool) – Random reverse input frames.

  • 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.

class basicsr.data.reds_dataset.REDSRecurrentDataset(opt)[source]

Bases: Dataset

REDS dataset for training recurrent networks.

The keys are generated from a meta info txt file. basicsr/data/meta_info/meta_info_REDS_GT.txt

Each line contains: 1. subfolder (clip) name; 2. frame number; 3. image shape, separated by a white space. Examples: 000 100 (720,1280,3) 001 100 (720,1280,3) …

Key examples: “000/00000000” GT (gt): Ground-Truth; LQ (lq): Low-Quality, e.g., low-resolution/blurry/noisy/compressed frames.

Parameters:

  • opt (dict) – Config for train dataset. It contains the following keys:

  • dataroot_gt (str) – Data root path for gt.

  • dataroot_lq (str) – Data root path for lq.

  • dataroot_flow (str, optional) – Data root path for flow.

  • meta_info_file (str) – Path for meta information file.

  • val_partition (str) – Validation partition types. ‘REDS4’ or ‘official’.

  • io_backend (dict) – IO backend type and other kwarg.

  • num_frame (int) – Window size for input frames.

  • gt_size (int) – Cropped patched size for gt patches.

  • interval_list (list) – Interval list for temporal augmentation.

  • random_reverse (bool) – Random reverse input frames.

  • 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.