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