sup3r.preprocessing.batch_queues.utilities.smooth_data

Contents

sup3r.preprocessing.batch_queues.utilities.smooth_data#

smooth_data(low_res, training_features, smoothing_ignore, smoothing=None)[source]#

Smooth data using a gaussian filter

Parameters:

  • low_res (Union[np.ndarray, da.core.Array]) – 4D | 5D array (batch_size, spatial_1, spatial_2, features) (batch_size, spatial_1, spatial_2, temporal, features)

  • training_features (list | None) – Ordered list of training features input to the generative model

  • smoothing_ignore (list | None) – List of features to ignore for the smoothing filter. None will smooth all features if smoothing kwarg is not None

  • smoothing (float | None) – Standard deviation to use for gaussian filtering of the coarse data. This can be tuned by matching the kinetic energy of a low resolution simulation with the kinetic energy of a coarsened and smoothed high resolution simulation. If None no smoothing is performed.

Returns:

low_res (Union[np.ndarray, da.core.Array]) – 4D | 5D array (batch_size, spatial_1, spatial_2, features) (batch_size, spatial_1, spatial_2, temporal, features)

Contents