SheppLoganDataset#

class deepinv.utils.phantoms.SheppLoganDataset(size=128, n_data=1, transform=None)[source]#

Bases: Dataset

Dataset for the single Shepp-Logan phantom. The dataset has length 1.

Parameters:

size (int) – Size of the phantom (square) image.
n_data (int) – Number of phantoms to generate per sample.
transform (Callable) – Transformation to apply to the output image.