python数据预处理

将图片按比例分在多个文件夹(7:3分成训练集和测试集)

• 目录结构
• 目的!：将image_region文件夹下面的01-12文件夹中的图像按比例分成训练集和测试集
• 格式要求：train文件夹中有文件夹为01-12，test文件夹中一样，其中的图片数据量不同

分之前目录结构

分之后目录结构

• 实现代码：

import os
import random
import shutil
from shutil import copy2

category = ['01', '02', '03', '04',
            '05', '06', '07', '08',
            '09', '10', '11', '12']
# 要保存的根目录
saveTrainPATH = './dataset/train'
saveTestPATH = './dataset/test'

if __name__ == "__main__":

    for j in range(12):
        # 子文件夹
        PATH = './image_region' + '/' + category[j]

        # 子文件夹路径
        for trainfiles in os.listdir(PATH):
            print(trainfiles)
            num_image = len(os.listdir(PATH))

            index_list = list(range(num_image))
            num = 0

            # 保存trian的路径-----------------------------
            trainDir = saveTrainPATH + '/' + category[j]
            # 先判断是否存在这个文件夹
            if not os.path.exists(trainDir):
                os.makedirs(str(trainDir))

            # 保存test的路径---------------------------------
            testDir = saveTestPATH + '/' + category[j]
            if not os.path.exists(testDir):
                os.makedirs(str(testDir))

            for i in index_list:
                fileName = os.path.join(PATH, os.listdir(PATH)[i])
                if num < num_image * 0.7:
                    copy2(fileName, trainDir)  # 复制过去,不改变原来目录的图片
                else:
                    copy2(fileName, testDir)
                num += 1

            print(trainDir, '\n', testDir)

参考：https://blog.csdn.net/weixin_44291388/article/details/106346379