deepinv.datasets

This module can be used for defining datasets or generating reconstruction datasets from other base datasets. Please refer to the user guide for more information.

Base Datasets

User Guide: refer to Base Datasets for more information.

deepinv.datasets.ImageDataset	Base class for imaging datasets in DeepInverse.
deepinv.datasets.ImageFolder	Dataset loading images from files.
deepinv.datasets.TensorDataset	Dataset wrapping data explicitly passed as tensors.

deepinv.datasets.check_dataset

Check that a torch dataset is compatible with DeepInverse.

Generating Datasets

User Guide: refer to Generating Datasets for more information.

deepinv.datasets.HDF5Dataset

DeepInverse HDF5 dataset with signal/measurement pairs (x, y).

deepinv.datasets.generate_dataset

Generates dataset of signal/measurement pairs from base dataset.

Image Datasets

User Guide: refer to Predefined Datasets for more information.

deepinv.datasets.DIV2K	Dataset for DIV2K Image Super-Resolution Challenge.
deepinv.datasets.Urban100HR	Dataset for Urban100 <https://paperswithcode.com/dataset/urban100>`_.
deepinv.datasets.Set14HR	Dataset for Set14.
deepinv.datasets.CBSD68	Dataset for CBSBD68.
deepinv.datasets.FastMRISliceDataset	Dataset for fastMRI that provides access to raw MR kspace data.
deepinv.datasets.SimpleFastMRISliceDataset	Simple FastMRI image dataset.
deepinv.datasets.CMRxReconSliceDataset	CMRxRecon dynamic MRI dataset.
deepinv.datasets.SKMTEASliceDataset	SKM-TEA dataset for raw multicoil MRI kspace data.
deepinv.datasets.LidcIdriSliceDataset	Dataset for LIDC-IDRI that provides access to CT image slices.
deepinv.datasets.Flickr2kHR	Dataset for Flickr2K.
deepinv.datasets.LsdirHR	Dataset for LSDIR.
deepinv.datasets.FMD	Dataset for Fluorescence Microscopy Denoising.
deepinv.datasets.Kohler	Dataset for Recording and Playback of Camera Shake
deepinv.datasets.NBUDataset	NBU remote sensing multispectral satellite imagery dataset.

Other Datasets

deepinv.datasets.PatchDataset	Builds the dataset of all patches from a tensor of images.
deepinv.datasets.utils.PlaceholderDataset	A placeholder dataset for test purposes.

Data Transforms

User Guide: refer to Data Transforms for more information.

deepinv.datasets.utils.Rescale	Image value rescale torchvision-style transform.
deepinv.datasets.utils.ToComplex	Torchvision-style transform to add empty imaginary dimension to image.
deepinv.datasets.utils.Crop	Torchvision-style transform to take crop in corner or any arbitrary place.
deepinv.datasets.MRISliceTransform	FastMRI raw data preprocessing.