SimpleFastMRISliceDataset#
- class deepinv.datasets.SimpleFastMRISliceDataset(root_dir, anatomy='knee', file_name=None, train=True, sample_index=None, train_percent=1.0, transform=None, download=False)[source]#
Bases:
Dataset
Simple FastMRI image dataset.
Loads in-memory a saved and processed subset of 2D slices from the full FastMRI slice dataset of Knoll et al.[1], for quick loading.
Important
By using this dataset, you confirm that you have agreed to and signed the FastMRI data use agreement.
These datasets are generated using
deepinv.datasets.FastMRISliceDataset.save_simple_dataset()
. You can use this to generate your own custom dataset and load using thefile_name
argument.We provide a pregenerated mini saved subset for singlecoil FastMRI knees (total 2 images) and RSS reconstructions of multicoil brains (total 2 images). These originate from their respective fully-sampled volumes converted to images via root-sum-of-square (RSS). Each slice is the middle slice from one independent volume. The images are of shape (2x320x320) and are normalised per-sample (0-1) and padded. Download the dataset using
download=True
, and load them using theanatomy
argument.Note
Since images are obtained from RSS, the imaginary part of each sample is 0.
- Examples:
Load mini demo knee dataset:
>>> from deepinv.datasets import SimpleFastMRISliceDataset >>> from deepinv.utils import get_data_home >>> dataset = SimpleFastMRISliceDataset(get_data_home(), anatomy="knee", download=True) >>> len(dataset) 2
- Parameters:
root_dir (str, pathlib.Path) – dataset root directory
anatomy (str) – load either fastmri “knee” or “brain” slice datasets.
file_name (str, pathlib.Path) – optional, name of local dataset to load, overrides
anatomy
. IfNone
, load dataset based onanatomy
parameter.train (bool) – whether to use training set or test set, defaults to True
sample_index (int) – if specified only load this sample, defaults to None
train_percent (float) – percentage train for train/test split, defaults to 1.
transform (Callable) – optional transform for images, defaults to None
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.
- References:
Examples using SimpleFastMRISliceDataset
:#

Self-supervised MRI reconstruction with Artifact2Artifact

Self-supervised learning with Equivariant Imaging for MRI.