stouputils.data_science.data_processing.image_augmentation module#

class ImageDatasetAugmentation(
final_dataset_size: int,
techniques: list[ProcessingTechnique] | None = None,
)[source]#

Bases: ImageDatasetPreprocess

Image dataset augmentation class. Check the class constructor for more information.

_add_suffix(
path: str,
used_destinations: set[str],
) → str[source]#

Add a count suffix to the path in order to avoid overwriting the same file

Parameters:

path (str) – Path to the file (example: “path/to/file.jpg”)

Returns:

Path with the suffix (example: “path/to/file_1.jpg”)

Return type:

str

get_queue(
dataset_path: str,
destination_path: str,
images_per_class_dict: dict[str, int] | None = None,
) → list[tuple[str, str, list[ProcessingTechnique]]][source]#

Get the queue of images to process with their techniques.

Parameters:

  • dataset_path (str) – Path to the dataset

  • destination_path (str) – Path to the destination dataset

  • images_per_class_dict (dict[str, int]) – Dictionary mapping class names to desired number of images (optional, defaults to empty dictionary)

Returns:

Queue of (source_path, dest_path, techniques) tuples

Return type:

list[tuple[str, str, list[ProcessingTechnique]]]