Rescale

class deepinv.datasets.utils.Rescale(*args, rescale_mode='min_max', **kwargs)

Bases: Module

Image value rescale torchvision-style transform.

The transform expects tensor of shape (…, H, W) and performs rescale over all dimensions (i.e. over all images in batch).

Parameters:

rescale_mode (str) – rescale mode, either “min_max” or “clip”.

forward(x)

Rescale image.

Parameters:

x (torch.Tensor) – image tensor of shape (…, H, W)