PlaceholderDataset

class deepinv.datasets.utils.PlaceholderDataset(n=1, shape=(1, 64, 64))

Bases: Dataset

A placeholder dataset for test purposes.

Produces image pairs x,y that are random tensor of shape specified.

Parameters:

• n (int) – number of samples in dataset, defaults to 1

• shape (tuple) – image shape, (channel, height, width), defaults to (1, 64, 64)