| 123456789101112131415161718192021222324252627282930313233343536 |
- from ipfml import processing
- from PIL import Image
- import shutil
- import os
- main_image_folder = "synthesis_images"
- images_folder = os.path.join(main_image_folder, "images")
- dest_folder = os.path.join(main_image_folder, "generated_blocks")
- if os.path.exists(dest_folder):
- # first remove folder if necessary
- os.rmdir(os.path.abspath(os.path.join(os.getcwd(), dest_folder)))
- # create new images
- images = os.listdir(images_folder)
- if not os.path.exists(dest_folder):
- os.makedirs(dest_folder)
- for img_path in images:
- img = Image.open(os.path.join(images_folder, img_path))
- blocks = processing.divide_in_blocks(img, (200, 200), pil=True)
- for id, pil_block in enumerate(blocks):
- img_name = img_path.split('/')[-1]
- split_name = img_name.split('.')
- block_name = split_name[0] + "_" + str(id) + "." + split_name[1]
- dest_img_block = os.path.join(dest_folder, block_name)
- pil_block.save(dest_img_block)
|
