Skip to content

Instantly share code, notes, and snippets.

@Diyago

Diyago/Setting generators.py

Created August 11, 2019 10:36
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 Diyago/cc26ccb0487dee0e5b3cc20ef332c0e1 to your computer and use it in GitHub Desktop.
Save Diyago/cc26ccb0487dee0e5b3cc20ef332c0e1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Setting generators.py
test_generator = DataGeneratorFolder(root_dir = './data/road_segmentation_ideal/training',
image_folder = 'input/',
mask_folder = 'output/',
nb_y_features = 1)
train_generator = DataGeneratorFolder(root_dir = './data/road_segmentation_ideal/training',
image_folder = 'input/',
mask_folder = 'output/',
batch_size=4,
image_size=512,
nb_y_features = 1, augmentation = aug_with_crop)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment