转载：voc数据集xml文件转换为txt文件并划分训练集、测试集

最新推荐文章于 2024-06-23 19:04:14 发布

小祥子ovo

最新推荐文章于 2024-06-23 19:04:14 发布

阅读量411

点赞数

文章标签： xml

原文链接：https://editor.csdn.net/md?not_checkout=1&articleId=118339863

版权

转载

感谢原作者:小白不白嘿嘿嘿
voc数据集xml文件转换为txt文件并划分训练集、测试集
附上我测试的代码：

import os
import shutil

root_path = r"C:/Users/18125/Desktop/transform/"
img_path = "C:/Users/18125/Desktop/transform/JPEGImages_insulator/"
txt_path = "C:/Users/18125/Desktop/transform/labels_insulator/"
def train_test_move(txt_path,root_path,img_path):
    files = os.listdir(txt_path)
    l = len(files)
    sets = ['train','val','test']
    k = 0
    p = 0.8
    for i in sets:
        if not os.path.exists(root_path+i):
            print(root_path+i)
            os.mkdir(root_path+i)
            os.mkdir(root_path+i+"/images")
            os.mkdir(root_path+i+"/labels")
        for file in files[round(l*k):round(l*p)]:
            shutil.copy(txt_path+file,root_path+i+"/labels")
            shutil.copy(img_path+file[:-3]+"jpg",root_path+i+"/images")
        k = p
        p += 0.1
train_test_move(txt_path,root_path,img_path)