Advertisement

python按照比例将文件夹中的数据集分割为训练数据和测试数据,并移动到其他文件夹中

阅读量:
复制代码
    ##深度学习过程中,需要制作训练集和验证集、测试集。
    import os, random, shutil
    def moveFile(fileDir):
        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 = "/home/nuaa301/yangshuang/MathModel/datasets3/train/300/"    #源图

全部评论 (0)

还没有任何评论哟~