当前位置:主页 > 软件编程 > Python代码 >

python实现对任意大小图片均匀切割的示例

时间:2022-04-10 10:32:50 | 栏目:Python代码 | 点击:

改代码是在windows 系统下

打开路径和保存路径换成自己的就可以啦~

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)

您可能感兴趣的文章:

相关文章