Skip to content

AutoNN.CNN.utils.image_augmentation.Augment

class Augment(path)
This class will augment the image dataset using the following image operations:

  • Rotations
  • Horizontal Flip
  • Vertical Flip

Parameters :

path : str | path to the image dataset folder

Tip
'''
    path: provide the path to your image folder
    which you want to augment
    ../Folder/dataset/cats/x1.png
    ../Folder/dataset/cats/x2.png
    .
    .
    .
    ../Folder/dataset/dogs/xx1.png
    ../Folder/dataset/dogs/xx2.png
    ../Folder/dataset/dogs/xx3.png
    .
    .
    path = '../Folder/dataset/'
'''

Method:

augment() :

Call this function to start augmentation

Parameters: None

Returns: None