aleju / imgaug

Image augmentation for machine learning experiments.
http://imgaug.readthedocs.io
MIT License
14.33k stars 2.43k forks source link

Number of images after data augmentation #774

Open denashamss opened 3 years ago

denashamss commented 3 years ago

Hi, I have applied the code below for data augmentation but I dont know how can I calculate the number of training dataset after data augmentation. augmentation = iaa.SomeOf((0, 2), [ iaa.Fliplr(0.5), iaa.Flipud(0.5), iaa.Affine(rotate=180), iaa.Multiply((0.1, 5.0)), iaa.GaussianBlur(sigma=(3.0, 10.0)), iaa.GammaContrast((2.0, 2.0)) ])

nonlinearjunkie commented 2 years ago
images_aug = augmentation(images=images)

Here, images_aug refers to batch output after augmentation of a batch of input images images. The number of image in images_aug is equal to the number of images in the batch images.