Number of images after data augmentation

[denashamss]

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]

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.