import os import random import shutil def moveFile(fileDir, saveDir): pathDir = os.listdir(fileDir) # 图片原始路径 filenumber = len(pathDir) rate1 = 0.2 # 自定义抽取csv文件的比例 picknumber1 = int(filenumber * rate1) # 按照rate比例从文件夹中取一定数量的文件 sample1 = random.sample(pathDir, picknumber1) # 随机选取 print(sample1) for name in sample1: shutil.move(fileDir + name, saveDir + "\\" + name) return if __name__ == '__main__': fileDir = "E:\管制物品检测数据集\images\\" #抽取图片文件夹下的路径 saveDir = 'E:\\test1' #保存被抽取图片的路径 moveFile(fileDir, saveDir)
按比例随机抽取文件夹下的图片,将抽取的图片移动到新的文件夹下
于 2022-08-31 13:18:43 首次发布