Set14HR

class deepinv.datasets.Set14HR(root: str, download: bool = False, transform: Callable | None = None)[source]

Bases: Dataset

Dataset for Set14.

The Set14 dataset is a dataset consisting of 14 images commonly used for testing performance of image reconstruction algorithms. Images have sizes ranging from 276×276 to 512×768 pixels.

Raw data file structure:

self.root --- Set14 --- image_SRF_2 --- img_001_SRF_2_bicubic.png
           |         |               |
           |         |               -- img_014_SRF_2_SRCNN.png
           |         |
           |         -- image_SRF_3 --- ...
           |         -- image_SRF_4 --- ...
           |
           -- Set14_SR.zip

This dataset wrapper gives access to the 14 high resolution images in the image_SRF_4 folder. Raw dataset source : https://github.com/jbhuang0604/SelfExSR

Parameters:
  • root (str) – Root directory of dataset. Directory path from where we load and save the dataset.

  • download (bool) – If True, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again. Default at False.

  • transform (callable, optional) – A function/transform that takes in a PIL image and returns a transformed version. E.g, torchvision.transforms.RandomCrop


Examples:

Instanciate dataset and download raw data from the Internet

>>> import shutil
>>> from deepinv.datasets import Set14HR
>>> dataset = Set14HR(root="Set14", download=True)  # download raw data at root and load dataset
Dataset has been successfully downloaded.
>>> print(dataset.check_dataset_exists())                # check that raw data has been downloaded correctly
True
>>> print(len(dataset))                                  # check that we have 14 images
14
>>> shutil.rmtree("Set14")                          # remove raw data from disk
check_dataset_exists() bool[source]

Verify that the image folders exist and contain all the images.

self.root should have the following structure:

self.root --- Set14 --- image_SRF_2 --- img_001_SRF_2_bicubic.png
           |         |               |
           |         |               -- img_014_SRF_2_SRCNN.png
           |         |
           |         -- image_SRF_3 --- ...
           |         -- image_SRF_4 --- ...
           |         -- xxx
           -- xxx