数据集的划分，简单实用

import os, random, shutil

def moveFile(fileDir, label):
    pathDir = os.listdir(fileDir)  # 取图片的原始路径
    filenumber = len(pathDir)
    rate = 0.2 # 自定义抽取图片的比例，比方说100张抽10张，那就是0.1
    picknumber = int(filenumber * rate)  # 按照rate比例从文件夹中取一定数量图片（如果不想设置比例，可以将rate = 0.2注释掉，直接自定义picknumber数值，如picknumber = 10）
    sample = random.sample(pathDir, picknumber)  # 随机选取picknumber数量的样本图片
    print(sample)
    for name in sample:
        # shutil.copy(fileDir + name, tarDir + name) # 复制
        shutil.move(fileDir + name, tarDir + name) # 剪切
        shutil.move(label + name[0:-3]+'txt', tarDir_label + name[0:-3]+'txt')  # 剪切
    return


if __name__ == '__main__':
    fileDir = "H:/original jujube/images/"  # 源图片文件夹路径
    tarDir = 'Jujube dataset/images/val/'  # 移动到新的文件夹路径
    label = 'H:/original jujube/labels/'    #标签位置
    tarDir_label = 'Jujube dataset/labels/val/' #移动路径
    moveFile(fileDir, label)