Python实现图片批量裁剪

from PIL import Image
import os

def crop_images(root_path, save_path):
    if not os.path.exists(save_path):
        os.makedirs(save_path)
        print(f"Folder '{save_path}' created successfully.")
    images = os.listdir(root_path)
    num = 0
    print("开始裁剪图片。。。。。。。。。。。。。。。。。。。。。")
    for image_name in images:
        if image_name.endswith(".jpg") or image_name.endswith(".png"):
            image_path = os.path.join(root_path, image_name)
            image = Image.open(image_path)
            rect = get_rect(image)
            crop_image = image.crop(rect)
            crop_image.save(os.path.join(save_path, image_name))
            if num == 0:
                print("原始图片分辨率为：", image.size)
            num += 1
            print("已裁剪", num, "张图片", "裁剪后图片分辨率为：", crop_image.size)
    print("裁剪完毕，共裁剪", num, "张图片")

def get_rect(image):
    width, height = image.size
    size = min(width, height)
    # (x,y,w,h), (x,y)表示裁剪图片的左上角, (w,h)表示在(x,y)左上角的位置开始，裁剪(w,h)大小的矩形区域.x表示水平方向,y表示垂直方向
    # rect = (0, 900, size, size+900)  # 4K分辨率，2160*3840
    # rect = (500, 0, size + 500, size)  # 4K分辨率，3840*2160
    rect = (420, 0, size+420, size)  # 1K分辨率，1920*1080
    # rect = (0, 420, size, size + 420)  # 1K分辨率，1080*1920
    return rect
#修改以下两个路径
root_path = "J:/XKQF5178"
save_path = "J:/XKQF5178/crop"
crop_images(root_path, save_path)