Skip to content

Instantly share code, notes, and snippets.

@um4ng-tiw
Created June 28, 2021 15:35
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save um4ng-tiw/2ae88d3cb31c271aaf426529d8ccca20 to your computer and use it in GitHub Desktop.
Save um4ng-tiw/2ae88d3cb31c271aaf426529d8ccca20 to your computer and use it in GitHub Desktop.
datagen = ImageDataGenerator(rescale = 1/255.0, validation_split = 0.25)
train_data_generator = datagen.flow_from_directory(directory="cell_images/train", target_size = (img_width, img_height),
class_mode="binary", batch_size = 16, subset = "training")
validation_data_generator = datagen.flow_from_directory(directory="cell_images/train", target_size = (img_width, img_height),
class_mode="binary", batch_size = 16, subset = "validation")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment