RandomPhantomDataset#

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

Bases: Dataset

Dataset of random ellipsoid phantoms generated on the fly.

Parameters:

  • length (float) – Length of the dataset.

  • 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.