Skip to content

Instantly share code, notes, and snippets.

@timehaven
Last active July 19, 2017 15:38
Show Gist options
  • Save timehaven/73e32c2d99285fe56e15e10a42c18852 to your computer and use it in GitHub Desktop.
Save timehaven/73e32c2d99285fe56e15e10a42c18852 to your computer and use it in GitHub Desktop.
train_datagen = ImageDataGenerator(...)
test_datagen = ImageDataGenerator(rescale=1. / 255)
train_generator = train_datagen.flow_from_directory(train_data_dir, ...)
validation_generator = test_datagen.flow_from_directory(validation_data_dir, ...)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment