Flickr2kHR#

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

Bases: Dataset

Dataset for Flickr2K.

Raw data file structure:

self.root --- Flickr2K --- 000001.png
           |            |
           |            -- 002650.png
           |
           -- Flickr2K.zip
Partial raw dataset source (only HR images) : https://huggingface.co/datasets/goodfellowliu/Flickr2K/resolve/main/Flickr2K.zip
Full raw dataset source (HR and LR images) : https://cv.snu.ac.kr/research/EDSR/Flickr2K.tar
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:

Instantiate dataset and download raw data from the Internet

from deepinv.datasets import Flickr2kHR
root = "/path/to/dataset/Flickr2K"
dataset = Flickr2kHR(root=root, download=True)  # download raw data at root and load dataset
print(dataset.check_dataset_exists())           # check that raw data has been downloaded correctly
print(len(dataset))                             # check that we have 100 images
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 --- Flickr2K --- 000001.png
           |            |
           |            -- 002650.png
           |
           -- xxx