DATASETS.md

June 1, 2021 · View on GitHub

Prepare Test Set

Prepare CUFED5 dataset

Download the CUFED5 dataset through this link.
Put the CUFED5 dataset into datasets folder. The folder structure should follow the format:

- CUFED5
|--- 000_0.png
|--- 000_1.png
|--- ...
|--- 125_4.png
|--- 125_5.png

Replace the following code into the dataset part of the corresponding configuration files.

datasets:
  test_1:
    name: CUFED5
    type: RefCUFEDDataset
    dataroot_in: ./datasets/CUFED5
    dataroot_ref: ./datasets/CUFED5
    io_backend:
      type: disk

    bicubic_model: PIL

    ann_file: ../datasets/CUFED5_pairs.txt

Prepare WR-SR dataset

Download the WR-SR dataset through this link.
Put the WR-SR dataset into datasets folder. The folder structure should follow the format:

- WR-SR
|-- input
|--- 001.png
|--- 002.png
|--- ...
|-- ref
|--- 001_ref.png
|--- 002_ref.png
|--- ...

Replace the following code into the dataset part of the corresponding configuration files.

datasets:
  test_1:
    name: WR-SR
    type: RefCUFEDDataset
    dataroot_in: ./datasets/WR-SR/input
    dataroot_ref: ./datasets/WR-SR/ref
    io_backend:
      type: disk

    bicubic_model: PIL

    ann_file: ./datasets/WR-SR_pairs.txt

Prepare Train Set

Prepare CUFED dataset

Download the CUFED dataset through this link.
Put the CUFED dataset into datasets folder. The folder structure should follow the format:

- CUFED
|-- input
|--- ****.png
|-- ref
|--- ****.png