拆分图片数据集的python

代码该怎么写？

可以使用Python中的PIL库来拆分图片数据集。以下是一个示例代码：

from PIL import Image
import os

def split_dataset(image_path, output_path, num_splits):
    # 打开图片
    img = Image.open(image_path)

    # 获取图片的宽和高
    width, height = img.size

    # 计算每个子图的宽和高
    split_width = width // num_splits
    split_height = height // num_splits

    # 循环创建子图
    for i in range(num_splits):
        for j in range(num_splits):
            # 计算子图的左上角和右下角坐标
            left = i * split_width
            top = j * split_height
            right = left + split_width
            bottom = top + split_height

            # 创建子图
            split_img = img.crop((left, top, right, bottom))

            # 保存子图
            split_img.save(os.path.join(output_path, f"{i}_{j}.jpg"))

这个函数接受三个参数：原始图片的路径、输出文件夹的路径和要拆分成的子图数量。它会将原始图片拆分成指定数量的子图，并将每个子图保存为一个单独的JPEG文件。