深度学习数据集划分：随机划分数据集

最新推荐文章于 2023-08-02 21:51:15 发布

ye1256

最新推荐文章于 2023-08-02 21:51:15 发布

阅读量360

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/ye1256/article/details/129612298

版权

# 数据集按比例随机划分   训练集  验证集

import os, random, shutil
def moveFile(fileDir):
        pathDir = os.listdir(fileDir)    #取图片的原始路径
        filenumber=len(pathDir)
        rate=0.5    #自定义抽取图片的比例，比方说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"C:\Users\yewenjing\Desktop\1104train\val1104 - 副本/"    #源图片文件夹路径
	tarDir = r'C:\Users\yewenjing\Desktop\1104train\test1104/'    #移动到新的文件夹路径
	moveFile(fileDir)