yolo切分数据集和图片

import os
import random
import shutil
def checkFile(path): #获取label的全部绝对路径存入列表
    listSaveLabel = []
    for i in os.listdir(path):
        listSaveLabel.append(path+"\\"+i)
    return listSaveLabel

def splitList(lst,ratio): #将列表按比例切分出来数据
    split_idx1 = int(len(lst) * ratio[0] / sum(ratio))
    split_idx2 = split_idx1 + int(len(lst) * ratio[1] / sum(ratio))
    return lst[:split_idx1], lst[split_idx1:split_idx2], lst[split_idx2:]

def saveLabel(saveLabelPath,train,val,test): #将切分出来的数据集进行数据划分
    trainPath = saveLabelPath+"//train"
    testPath = saveLabelPath+"//test"
    valPath = saveLabelPath+"//val"

    if not os.path.exists(testPath):
        os.makedirs(testPath)
    if not os.path.exists(trainPath):
        os.makedirs(trainPath)
    if not os.path.exists(valPath):
        os.makedirs(valPath)

    for trainFile in train:
        if os.path.isfile(trainFile):