spyrit.misc.statistics.mean_2

spyrit.misc.statistics.mean_2(dataloader: DataLoader, device: device, n_loop: int = 1)

Computes 2D mean image computed across batches and channels

Args:

dataloader (torch.utils.data.DataLoader): Dataloader. The fetch data are Torch tensors with shape (B,C,N,N).

device (torch.device): Device.

n_loop (int, optional): Number of loops across image database. Defaults to 1. nloop > 1 is relevant for dataloaders with random transforms.

Returns:

mean (np.array): Mean image with shape (N, N).