'StratifiedKfold in ImageDataGenerator flow_from_directory

Is there any solution to implement CrossValidation directly from ImageDataGenerator? I have a folder containing all my training images (and masks since I'm performing segmentation task).

I want to perform StratifiedKfold directly from ImageDatagenerator flow_from_directory. Any ideas?



Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source