deepinv.datasets#

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

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.

deepinv.datasets.Set14HR

Dataset for Set14.

deepinv.datasets.CBSD68

Dataset for CBSBD68.

deepinv.datasets.FastMRISliceDataset

Dataset for fastMRI that provides access to raw MR image slices.

deepinv.datasets.SimpleFastMRISliceDataset

Simple FastMRI image dataset.

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.