import numpy as npimport matplotlibimport 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)
