Skip to content

Instantly share code, notes, and snippets.

@Karts27

Karts27/load_image_generator.py Secret

Last active November 16, 2021 16:27
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 Karts27/760023c0ccd8a66ede4497c41df1fe8a to your computer and use it in GitHub Desktop.
Save Karts27/760023c0ccd8a66ede4497c41df1fe8a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_image_generator.py
batch_size = 20
training_images = train_image_generator.flow_from_directory(train_path,
target_size = (64,64),
class_mode = 'categorical',
batch_size = batch_size,
classes = os.listdir(train_path),
subset = 'training',
shuffle = True
#color_mode="grayscale"
)
validation_images = train_image_generator.flow_from_directory(train_path,
target_size = (64,64),
class_mode = 'categorical',
batch_size = batch_size,
classes = os.listdir(train_path),
subset = 'validation',
shuffle = True
#color_mode="grayscale"
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment