from re import I
import shutil 
import os 


image_path = '/home/admin-root/haotian/锻8/tensorrtx/yolov8/images_20250624105811'
label_path = '/home/admin-root/haotian/python哈汽锻8安全帽识别/export/duan8_11/labels'
destination_image_path = '/home/admin-root/haotian/python哈汽锻8安全帽识别/DatasetHelmetHeadShoe/train/images'
destination_label_path = '/home/admin-root/haotian/python哈汽锻8安全帽识别/DatasetHelmetHeadShoe/train/labels'


image_name = [ t.split('.')[0] for t in  os.listdir(label_path)]

for i in range(len(image_name)):
    # 复制图片文件
    shutil.copy2(os.path.join(image_path, image_name[i]+'.jpg'), os.path.join(destination_image_path, f'22_{i}.jpg'))
    
    # 复制标签文件
    shutil.copy2(os.path.join(label_path, image_name[i]+'.txt'), os.path.join(destination_label_path, f'22_{i}.txt'))
    
    print(f"{i} finished")