【工具】python 按照比例划分训练测试集

import shutil
from glob import glob
import os
import random
from tqdm import tqdm

path1 = "path1" #数据路径
paths = sorted(glob(path1+"/*.jpg"))
savepath = "test"
os.makedirs(savepath,exist_ok=True)
testpaths = random.sample(paths, int(len(paths)*0.2))
for path in tqdm(testpaths):
    shutil.move(path,savepath)
    shutil.move(path.replace(".jpg",".txt"),savepath)