数据集扩充方法（图像分割成几部分）

Posted 2022-12-01 告白少年

解决图像数据集不足
将数据集每个图像分割成几部分（下面例子是把一张图像分为9部分，根据自己需要修改参数）

from PIL import Image
import sys
import os


def cut_image(image):
    width, height = image.size
    item_width = int(width /3)#修改这里的参数
    item_height=int(height / 3)#修改这里的参数
    box_list = []
    count = 0
    for j in range(0,3):#修改这里的参数
        for i in range(0, 3):#修改这里的参数
            count += 1
            box = (i * item_width, j * item_height, (i + 1) * item_width, (j + 1) * item_height)
            box_list.append(box)
    print(count)
    image_list = [image.crop(box) for box in box_list]
    return image_list


def save_images(image_list,image_name,save_path):
    index = 1
    for image in image_list:
        image.save(os.path.join(save_path,image_name.split(".")[0]+str(index)+".png"))
        index += 1


if __name__ == '__main__':
    file_path = r"F:\\h_package\\data_bright\\over_exposure\\over"
    save_path=r"F:\\h_package\\data_bright\\over_exposure\\over_new"
    # 打开图像
    for image_name in os.listdir(file_path):
        print(image_name)
        image = Image.open(os.path.join(file_path,image_name))
        image_list = cut_image(image)
        save_images(image_list,image_name,save_path)