Is there any replacement in Julia for ImageDataGenerator

General Usage
9

Try

using PyCall
ImageDataGenerator = pyimport("keras").preprocessing.image.ImageDataGenerator
train = ImageDataGenerator(rescale = 1/255)
train_generator = train.flow_from_directory(train_dir, target_size = (150, 150), batch_size = 20, class_mode = "binary")

I have no idea how easily train_generator can be hooked up to Flux but at least you have it available on the Julia side of PyCall.

1 Like