import os
import shutil

d_train_image_file = 'newdataset/train/images'
d_train_label_file = 'newdataset/train/labels'

d_val_image_file = 'newdataset/val/images'
d_val_label_file = 'newdataset/val/labels'

s_train_image1_file = 'HelmetDetection/train/images'
s_train_label1_file = 'HelmetDetection/train/labels'

s_train_image2_file = 'fire_smoke/train/images'
s_train_label2_file = 'fire_smoke/train/labels'

s_train_image3_file = 'fire-8/train/images'
s_train_label3_file = 'fire-8/train/newlabels'

s_val_image1_file = 'HelmetDetection/val/images'
s_val_label1_file = 'HelmetDetection/val/labels'

s_val_image2_file = 'fire_smoke/val/images'
s_val_label2_file = 'fire_smoke/val/labels'

s_val_image3_file = 'fire-8/valid/images'
s_val_label3_file = 'fire-8/valid/newlabels'


s_train_image = [s_train_image1_file, s_train_image2_file, s_train_image3_file]
s_train_label = [s_train_label1_file, s_train_label2_file, s_train_label3_file]

s_val_image = [s_val_image1_file, s_val_image2_file, s_val_image3_file]
s_val_label = [s_val_label1_file, s_val_label2_file, s_val_label3_file]


# train file copy
# for i in range(3):
#     all_train_image_name = os.listdir(s_train_image[i])
#     all_train_label_name = os.listdir(s_train_label[i])
#     for j in range(len(all_train_image_name)):
#         s_image_path = os.path.join(s_train_image[i], all_train_image_name[j])
#         d_image_path = os.path.join(d_train_image_file, all_train_image_name[j])
#
#         s_label_path = os.path.join(s_train_label[i], all_train_label_name[j])
#         d_label_path = os.path.join(d_train_label_file, all_train_label_name[j])
#
#         try:
#             shutil.copy(s_image_path, d_image_path)
#             shutil.copy(s_label_path, d_label_path)
#             print('finished')
#         except Exception as e:
#             print(f"复制文件时出错: {e}")

# val
for i in range(3):
    all_val_image_name = os.listdir(s_val_image[i])
    all_val_label_name = os.listdir(s_val_label[i])
    for j in range(len(all_val_image_name)):
        s_image_path = os.path.join(s_val_image[i], all_val_image_name[j])
        d_image_path = os.path.join(d_val_image_file, all_val_image_name[j])

        s_label_path = os.path.join(s_val_label[i], all_val_label_name[j])
        d_label_path = os.path.join(d_val_label_file, all_val_label_name[j])

        try:
            shutil.copy(s_image_path, d_image_path)
            shutil.copy(s_label_path, d_label_path)
            print('finished')
        except Exception as e:
            print(f"复制文件时出错: {e}")