将图片按照指定大小批量进行裁剪(可设置步长_python)

将图片按照指定大小批量进行裁剪(可设置步长_python)

import os
from PIL import Image

# 设置更高的图像大小限制，禁用解压炸弹检查
Image.MAX_IMAGE_PIXELS = None  # 禁用解压炸弹检查

def crop_image(image_path, block_size=(640, 640), step_size=(340, 340)):
    # 打开图片
    img = Image.open(image_path)
    
    # 获取原图的宽度和高度
    img_width, img_height = img.size

    # 计算裁剪块的宽度和高度
    block_width, block_height = block_size

    # 确保图片尺寸可以被整除（如果不能整除，可以裁剪多余部分）
    cropped_images = []

    for top in range(0, img_height, step_size[1]):  # 使用step_size[1]作为纵向步长
        for left in range(0, img_width, step_size[0]):  # 使用step_size[0]作为横向步长
            # 确保裁剪区域不会越界
            right = min(left + block_width, img_width)
            bottom = min(top + block_height, img_height)
            
            # 裁剪图片
            cropped_image = img.crop((left, top, right, bottom))
            cropped_images.append(cropped_image)

    return cropped_images

def crop_images_in_folder(input_folder, output_folder, block_size=(640, 640), step_size=(340, 340)):
    # 如果输出文件夹不存在，则创建它
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    
    # 遍历输入文件夹中的每一张图片
    for filename in os.listdir(input_folder):
        # 检查文件是否为图片（根据文件扩展名）
        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
            image_path = os.path.join(input_folder, filename)
            cropped_images = crop_image(image_path, block_size, step_size)

            # 为每个裁剪的图像保存一个新文件
            for i, cropped_img in enumerate(cropped_images):
                output_path = os.path.join(output_folder, f"{os.path.splitext(filename)[0]}_cropped_{i}.jpg")
                cropped_img.save(output_path)
                print(f"保存裁剪图像: {output_path}")

# 输入文件夹和输出文件夹路径
input_folder = r"D:\BaiduNetdiskDownload\数据"  # 请替换成你的输入文件夹路径
output_folder = r"D:\BaiduNetdiskDownload\图片裁剪"  # 请替换成你的输出文件夹路径
block_size = (640, 640)  # 每个小块的尺寸
step_size = (340, 340)  # 步长设置为340

# 执行裁剪操作
crop_images_in_folder(input_folder, output_folder, block_size, step_size)