Skip to content

Instantly share code, notes, and snippets.

@ParthNipunDave

ParthNipunDave/zoo_data_generators.py

Created June 7, 2021 12:12
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ParthNipunDave/93bcc24cb1031127d3de11d5c6190484 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ParthNipunDave/93bcc24cb1031127d3de11d5c6190484 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
zoo_data_generators.py
data=ImageDataGenerator(validation_split=0.3,zoom_range=0.3,rescale=1./255,horizontal_flip=True)
train_data=data.flow_from_directory('../input/animals-detection-images-dataset/train',target_size=(255,255),subset='training',shuffle=True,batch_size=8)
valid_data=data.flow_from_directory('../input/animals-detection-images-dataset/train',target_size=(255,255),subset='validation',shuffle=True,batch_size=8)
batch_size=8
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment