avocado.Augmentor.augment_dataset

Augmentor.augment_dataset(augment_name, dataset, num_augments, include_reference=True)

Generate augmented versions of all objects in a dataset.

Parameters:
  • augment_name (str) – The name of the augmented dataset.
  • dataset (Dataset) – The dataset to use as a reference for the augmentation.
  • num_augments (int) – The number of times to use each object in the dataset as a reference for augmentation. Note that augmentation sometimes fails, so this is the number of tries, not the number of sucesses.
  • include_reference (bool (optional)) – If True (default), the reference objects are included in the new augmented dataset. Otherwise they are dropped.
Returns:

augmented_dataset – The augmented dataset.

Return type:

Dataset