moonshine.preprocessing.get_preprocessing_fn

moonshine.preprocessing.get_preprocessing_fn(model: str, dataset: str) → Callable

Get a preprocessing function for a given model and dataset. For a full list of preprocessing functions available, see the documentation.

Args:

model: Which type of model to preprocess for, e.g. unet. dataset: Which dataset to expect for preprocessing, e.g. fmow_rgb.

Returns:

fn: A function that can be applied to an input array.