Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
datagen = ImageDataGenerator(rescale = 1/255.0, validation_split = 0.2)
train_data_generator = datagen.flow_from_directory(directory="Blood_cell_image_dataset/images/TRAIN/",
target_size = (img_width, img_height), color_mode="rgb",
class_mode="categorical", batch_size = 16, shuffle=True ,subset = "training")
validation_data_generator = datagen.flow_from_directory(directory="Blood_cell_image_dataset/images/TRAIN/",
target_size = (img_width, img_height), color_mode="rgb",
class_mode="categorical", batch_size = 16, shuffle=True, subset = "validation")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment