After Image Augmentation, labels are messed up! Pytorch

1

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.

Screen Shot 2020-12-05 at 5.07.25 PM
Screen Shot 2020-12-05 at 5.07.25 PM816×809 117 KB

2

The augmented images do not have correct labels.

Screen Shot 2020-12-05 at 5.03.19 PM
Screen Shot 2020-12-05 at 5.03.19 PM1433×263 115 KB

3

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.