My dataset includes 7 classes with .png images The generator worked well when they find 10003 images and 7 classes but when I ouput the first image from train_gen it was all