Skip to content

Instantly share code, notes, and snippets.

@Tob-iee
Last active February 7, 2024 10:09
Show Gist options
  • Save Tob-iee/bb867e762d31a633acb66924fc8d6684 to your computer and use it in GitHub Desktop.
Save Tob-iee/bb867e762d31a633acb66924fc8d6684 to your computer and use it in GitHub Desktop.
# create training and validation data generators
IMG_SIZE = (224,224)
BATCH_SIZE = 8
def train_val_generators(train_dir, val_dir):
train_gen = ImageDataGenerator(rescale=1/255.,
horizontal_flip=True,
vertical_flip=True,
brightness_range=(0, 0.2),
rotation_range=0.2,
zoom_range=0.2,)
val_gen = ImageDataGenerator(rescale=1/255.)
# create image generators to feed images to the model
train_generator, val_generator = train_val_generators(train_dir, val_dir)
train_generator = train_gen.flow_from_directory(directory=train_dir,
batch_size=BATCH_SIZE,
class_mode="categorical",
target_size=IMG_SIZE)
val_generator = val_gen.flow_from_directory(directory=val_dir,
batch_size=BATCH_SIZE,
class_mode="categorical",
target_size=IMG_SIZE)
return train_generator, val_generator
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment