src.ml.dataset.calculate_mean_std

calculate_mean_std(dataset)[source]

Calculate dataset mean and standard deviation.

Parameters

dataset – PyTorch dataset object.

Returns

A tuple that consists a tuple of mean and std for 3 channels (mean, std).

Return type

tuple