python实现数据标注的分配,分配后的目录层级如下
import os
import shutil
import glob
from tqdm import tqdm
import random
# 数据文件夹路径
source_folder = 'data'
# 需要分配的日期和人
name=["Nancy/","Jerry/","Tom/","Ben/","Mike/"]
date=["20250107","20250108","20250109","20250110"]
# 每人每天需要标注的数据量
each_person_one_day = 10
# 获取数据文件夹中的所有图片
images = glob.glob(os.path.join(source_folder, '*.png')) # 假设是png图片,可以根据需要修改扩展名
# 打乱文件顺序
random.shuffle(images)
# 开始分配
for name_id in range(len(name)):
for date_id in range(len(date)):
for i in tqdm(images[name_id*len(date)*each_person_one_day+date_id*each_person_one_day:name_id*len(date)*each_person_one_day+(date_id+1)*each_person_one_day], desc=f'Copying data for {name[name_id]+date[date_id]}'):
src = i
os.makedirs(name[name_id]+date[date_id], exist_ok=True)
dst = os.path.join(name[name_id]+date[date_id], os.path.basename(src))
shutil.copy2(src, dst)