怎么办_怎么做_How to do anything_解决问题万事OK

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 = "./source/"    #源图片文件夹路径
    tarDir = './result/'    #移动到新的文件夹路径
    moveFile(fileDir)
 
【关键字】:

此文由 yeaso   万事OK网编辑:首页 > IT » python从原始数据集制作验证集训练集随机抽取一定数量的图片

相关推荐

© 2020 怎么办_怎么做_How to do anything_解决问题万事OK  京ICP备13039979号-5