转载:https://blog.csdn.net/qq_33188180/article/details/109237401
代码如下:
import os,random,shutil
def moveFile(fileDir, tarDir):
pathDir = os.listdir(fileDir) # 取图片的原始路径
filenumber = len(pathDir)
rate = 0.11 # 自定义抽取图片的比例,比方说100张抽10张,那就是0.1
picknumber = int(filenumber * rate) # 按照rate比例从文件夹中取一定数量图片
sample = random.sample(pathDir, picknumber) # 随机选取picknumber数量的样本图片
print(sample)
print(len(sample))
for name in sample:
shutil.move(fileDir + name, tarDir + name)
if __name__ == '__main__':
fileDir = "/home/xiaobumidm/Yolo_mark-master/验证/训练集/明显/" # 源图片文件夹路径
tarDir = '/home/xiaobumidm/Yolo_mark-master/验证/验证集/明显/' # 移动到新的文件夹路径
moveFile(fileDir, tarDir)