I would like to ask you how to solve this problem. I have two folders of pictures, one as train set and the other as validation. What i\'ve done is to use ImageDataGenerator: