After Image Augmentation, labels are messed up! Pytorch

Hi everone, :smiley:I am doing a project on cat breeds classification. As I am doing data augmentation, the number of images is doubled but the labels do not match the augmented images. Here is my code.

The augmented images do not have correct labels.

The number of images wonโ€™t change, since the transformations are applied on the fly in the __getitem__ method, so I would recommend to recheck, if the image folders contain unexpected files.