How to get sample size with/without data augmentation

[td43]

Hi,

I would like to know how to get the number of samples with and without data_augmentation since I need to check that actually the dataset has been augmented.

I have tried to get the len() from the dataset in InputReader class in dataloader.py but it seems does not work for a tfrecord_file. Also, I have used the cardinality function of tf2