按图像尺寸平均切分图像

思路：

1. 首先，把图像填充为正方形（以图像的最长边为正方形新图边长，对较短边长居中填充空白）
2. 循环迭代切分图像
3. 保存生成的图像列表

from PIL import Image
import sys
import os

# 将图片填充为正方形
def fill_image(image):
    width, height = image.size
    # 选取长和宽中较大值作为新图片的
    new_image_length = width if width > height else height
    # 生成新图片[白底]
    new_image = Image.new(image.mode, (new_image_length, new_image_length), color='white')
    # 将之前的图粘贴在新图上，居中
    if width > height:  # 原图宽大于高，则填充图片的竖直维度
        # (x,y)二元组表示粘贴上图相对下图的起始位置
        new_image.paste(image, (0, int((new_image_length - height) / 2)))
    else:
        new_image.paste(image, (int((new_image_length - width) / 2), 0))
    return new_image


# 平均切四份图
def cut_image(image):
    width, height = image.size
    item_width = int(width / 2)
    box_list = []
    # (left, upper, right, lower)
    for i in range(0, 2):  # 两重循环，生成4张图片基于原图的位置
        for j in range(0, 2):
            # print((i*item_width,j*item_width,(i+1)*item_width,(j+1)*item_width))
            box = (j * item_width, i * item_width, (j + 1) * item_width, (i + 1) * item_width)
            box_list.append(box)

    image_list = [image.crop(box) for box in box_list]
    return image_list


# 保存
def save_images(image_list, saveDir, name): # image_list:图像数组 ， savePath： 图像保存目录， name：新的图像名
    index = 1
    for image in image_list:
        image.save(os.path.join(saveDir,str(index)+name))
        index += 1


if __name__ == '__main__':
    sourcepath = 'E:/图像集/图像集'
    dirPath = 'img'
    for sampledir in os.listdir(sourcepath):
        # valClassNum += 1
        imageDir = sourcepath + "/" + sampledir
        dirPath = dirPath + '/' + sampledir
        for name in os.listdir(imageDir):
            file_path = imageDir + "/" + name
            image = Image.open(file_path)
            # image.show()
            image = fill_image(image)
            image_list = cut_image(image)
            print(dirPath)
            save_images(image_list, dirPath, name)
        dirPath = 'img'