70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
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}") |