kymatio
kymatio copied to clipboard
Sample to use Scattering 2D with ImageDataGenerator.flowfromdirectory
testDataGen = ImageDataGenerator(rescale = 1.0 / 255)
trainGenerator = trainDataGen.flow_from_directory(os.path.join(dspth,'SplitDataset', 'Train'), target_size = (32, 32), batch_size = 32, color_mode = 'grayscale', classes = [str(Class) for Class in [15,27]], class_mode = 'categorical')
validationGenerator = testDataGen.flow_from_directory(os.path.join(dspth,'SplitDataset', 'Validation'), target_size = (32, 32), batch_size = 32, color_mode = 'grayscale', classes = [str(Class) for Class in [15,27]], class_mode = 'categorical') model.fit(trainGenerator, epochs = 40, validation_data =validationGenerator, callbacks = callbacks)