21 lines
885 B
Python
21 lines
885 B
Python
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") |