python-5000*5000子图分割遥感图像

对超大遥感图像进行分割，每张子图的大小为5000*5000像素

from PIL import Image
# 解决文件过大问题
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
Image.MAX_IMAGE_PIXELS = None
import sys


# 先将 input image 填充为正方形
def fill_image(image):
    width, height = image.size
    # 选取长和宽中较大值作为新图片的
    new_image_length = width if width > height else height
    # 定义边长为5000的倍数，将图像作为5000像素的方块整数裁剪，cuttimes为每个边上切割的次数
    cut_times = int(new_image_length / 5000) + 1
    new_image_length = cut_times * 5000
    # 生成新图片[白底]
    new_image = Image.new(image.mode, (new_image_length, new_image_length), color='white')  # 注意这个函数！
    # 将原来的图像贴在新的白底上,粘贴位置从左上角（0，0）处开始
    new_image.paste(image, (0, 0))

    return new_image


def cut_image(image):
    width, height = image.size
    # 定义新的子图大小为5000像素
    item_width = 5000
    # 计算每个边上切割的次数cut_times
    cut_times = (int(width / 5000) + 1) if width > height else (int(height / 5000) + 1)

    box_list = []
    # (left, upper, right, lower)
    for i in range(0, cut_times-1):
        for j in range(0, cut_times-1):
            # 打印切割后尺寸
            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):
    index = 1
    for image in image_list:
        image.save(str(index) + '.png', 'PNG')
        index += 1


if __name__ == '__main__':
    file_path = r"G:\DataSets\satellite-images-from-school\images\wv卫星 0.3米样例.jpg"  # 图片保存的地址
    image = Image.open(file_path)
    # image.show()
    image_new = fill_image(image)
    image_list = cut_image(image_new)
    save_images(image_list)