python实现从一个文件夹里随机提取特定比例文件放到另一文件夹中,训练验证集划分

效果:
python实现从一个文件夹里随机提取特定比例文件放到另一文件夹中,训练验证集划分_第1张图片

import os, random, shutil


def moveFile(fileDir, tarDir):
    pathDir = os.listdir(fileDir)  # 取图片的原始路径
    filenumber = len(pathDir)
    rate = 0.1  # 自定义抽取图片的比例,比方说100张抽10张,那就是0.1
    picknumber = int(filenumber * rate)  # 按照rate比例从文件夹中取一定数量图片
    sample = random.sample(pathDir, picknumber)  # 随机选取picknumber数量的样本图片
    print(sample)
    for name in sample:
        shutil.move(fileDir + name, tarDir + "\\" + name)
    return

if __name__ == '__main__':
    fileDir = r"E:\01_hjz\datas\facemask\mask10000_new\images\train" + "\\"  # 源图片文件夹路径
    tarDir = r'E:\01_hjz\datas\facemask\mask10000_new\images\val'  # 移动到新的文件夹路径
    moveFile(fileDir, tarDir)

你可能感兴趣的:(Python)