Guest User

Untitled

a guest
Feb 23rd, 2018
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.00 KB | None | 0 0
  1. # we create two instances with the same arguments
  2. data_gen_args = dict(featurewise_center=True,
  3. featurewise_std_normalization=True,
  4. rotation_range=90.,
  5. width_shift_range=0.1,
  6. height_shift_range=0.1,
  7. zoom_range=0.2)
  8. image_datagen = ImageDataGenerator(**data_gen_args)
  9. mask_datagen = ImageDataGenerator(**data_gen_args)
  10.  
  11. # Provide the same seed and keyword arguments to the fit and flow methods
  12. seed = 1
  13. image_datagen.fit(images, augment=True, seed=seed)
  14. mask_datagen.fit(masks, augment=True, seed=seed)
  15.  
  16. image_generator = image_datagen.flow_from_directory(
  17. 'data/images',
  18. class_mode=None,
  19. seed=seed)
  20.  
  21. mask_generator = mask_datagen.flow_from_directory(
  22. 'data/masks',
  23. class_mode=None,
  24. seed=seed)
  25.  
  26. # combine generators into one which yields image and masks
  27. train_generator = zip(image_generator, mask_generator)
  28.  
  29. model.fit_generator(
  30. train_generator,
  31. steps_per_epoch=2000,
  32. epochs=50)
Add Comment
Please, Sign In to add comment