import numpy as np
import matplotlib
import os
 
def img_seg(dir):
  files = os.listdir(dir)
  for file in files:
    a, b = os.path.splitext(file)
    img = Image.open(os.path.join(dir + "\\" + file))
    hight, width = img.size
    w = 256
    id = 1
    i = 0
    while (i + w <= hight):
      j = 0
      while (j + w <= width):
        new_img = img.crop((i, j, i + w, j + w))
        #rename = "D:\\labelme\\images\\"
        rename = "D:\\labelme\\annotations\\"
        new_img.save(rename + a + "_" + str(id) + b)
        id += 1
        j += w
      i = i + w
 
 
if __name__ == '__main__':
  #path = "D:\\labelme\\data\\images\\train"
  path = "D:\\labelme\\data\\dataset_png"
  img_seg(path)