stouputils.data_science.scripts.augment_dataset module#

augment_dataset(
techniques: list[ProcessingTechnique],
default_type: Literal['image'] = 'image',
default_input: str = '/home/runner/work/stouputils/stouputils/data/hip_implant',
default_output: str = '',
default_final_dataset_size: int = 1000,
) → None[source]#

Augment a dataset with various data processing techniques.

This script takes a dataset path and applies configurable processing techniques to generate an expanded dataset. The augmented data is saved to a destination path. The augmentation can be done for images or other data types.

Parameters:

  • default_type (str) – Default type of data to augment.

  • default_input (str) – Default path to the input dataset.

  • default_output (str) – Default path to save the augmented dataset.

  • default_final_dataset_size (int) – Default final size of the dataset.

Returns:

This function does not return anything.

Return type:

None