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)