generate_dataset.py 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #!/usr/bin/env python2
  2. # -*- coding: utf-8 -*-
  3. """
  4. Created on Fri Sep 14 21:02:42 2018
  5. @author: jbuisine
  6. """
  7. from __future__ import print_function
  8. import keras
  9. from keras.datasets import cifar10
  10. from keras.preprocessing.image import ImageDataGenerator
  11. from keras.models import Sequential
  12. from keras.layers import Dense, Dropout, Activation, Flatten
  13. from keras.layers import Conv2D, MaxPooling2D
  14. import os, glob, image_slicer
  15. from PIL import Image
  16. # show to create own dataset https://gist.github.com/fchollet/0830affa1f7f19fd47b06d4cf89ed44d
  17. NUMBER_SUB_IMAGES = 1600
  18. def create_images(folder, output_folder):
  19. images_path = glob.glob(folder + "/*.png")
  20. for img in images_path:
  21. image_name = img.replace(folder, '').replace('/', '')
  22. tiles = image_slicer.slice(img, NUMBER_SUB_IMAGES, save = False)
  23. image_slicer.save_tiles(tiles, directory=output_folder, prefix='part_'+image_name)
  24. def generate_dataset():
  25. create_images('img_train/final', 'data/train/final')
  26. create_images('img_train/noisy', 'data/train/noisy')
  27. create_images('img_validation/final', 'data/validation/final')
  28. create_images('img_validation/noisy', 'data/validation/noisy')
  29. def main():
  30. # create database using img folder (generate first time only)
  31. generate_dataset()
  32. if __name__== "__main__":
  33. main()