Based on your code each image will be randomly augmented in the epoch.
If you want to return the same image using different augmentations (and thus increase the number of samples in the epoch) you could just call self.aug multiple times and return all augmented images.