NEU-DET数据集， xml转txt . XML转TXT

首先是XML转TXT

 

import pathlib
import xml.etree.ElementTree as ET#xml 是python自带的package
import os
from pathlib import Path


classes=["crazing", "inclusion", "patches", "pitted_surface", "rolled-in_scale", "scratches"]#写自己的分类名,有几个写几个
pre_dir=Path(r'F:\Project\Project2023.6.23\NEU-DET\ANNOTATIONS')#xml文件所在文件夹F:\Project\Project2023.6.23\NEU-DET\ANNOTATIONS
target_dir=Path(r'F:\Project\Project2023.6.23\target_sfid\labels\new')#想要存储txt文件的文件夹
# path=os.listdir(pre_dir)

for path in pre_dir.iterdir():
# path1=r'C:\Users\loadlicb\Desktop\chrome_RTJOXXsYHM.xml'#xml文件路径
    tree=ET.parse(path)
    root=tree.getroot()#这两个步骤将xml文件拆出来了
    oo=[]
    for child in root:
        if child.tag == 'filename':#tag对应的是《》中的内容，text对应的是两个《》中间的部分内容
            oo.append(child.text)#获得xml文件的名
            # print(child.text)
        for i in child:

            if i.tag == 'width':#获得图片的w
                oo.append(i.text)
                # print(i.text)
            if i.tag == 'height':#获得图片的h
                oo.append(i.text)
                # print(i.text)
            if i.tag == 'name':#获得当前框的class
                oo.append(i.text)
                # print(i.text)

            for j in i:
                if j.tag == 'xmin':#获得当前框的两个对角线上的点的两组坐标
                    oo.append(j.text)
                    # print(j.text)
                if j.tag == 'ymin':
                    oo.append(j.text)
                    # print(j.text)
                if j.tag == 'xmax':
                    oo.append(j.text)
                    # print(j.text)
                if j.tag == 'ymax':
                    oo.append(j.text)
                    # print(j.text)
    # print(oo)
    filename=oo[0]#读取图片的名和宽高
    filename=os.path.split(filename)
    # print(filename)
    name,extension=os.path.splitext(filename[1])#获取xml名和后缀
    width=oo[1]
    dw=1/int(width)
    height=oo[2]
    dh=1/int(height)
    oo.pop(0)
    oo.pop(0)
    oo.pop(0)#删除三次oolist的0号元素


    back=[]
    # print((len(oo))%5)
    for i in range(len(oo)//5):
        for p in range(len(classes)):#划定class的序号
            if classes[p] == oo[5*i]:#str == str
                cl=p
                back.append(cl)
        x=(int(oo[5*i+1])+int(oo[5*i+3]))/2#oo里的所有元素都是str，数字也是
        y = (int(oo[5 * i + 2]) + int(oo[5 * i + 4])) / 2#计算标注框的中心点的xy坐标
        w=int(oo[5*i+3])-int(oo[5*i+1])
        h=int(oo[5*i+4])-int(oo[5*i+2])#计算标注框的宽高
        back.append('{:.4f}'.format(x*dw))
        back.append('{:.4f}'.format(y * dh))
        back.append('{:.4f}'.format(w * dw))
        back.append('{:.4f}'.format(h * dh))
        # back.append(y*dh)
        # back.append(w*dw)
        # back.append(h*dh)#转换到0-1区间
    # print(back)
    # dir=r'C:\Users\loadlicb\Desktop'#label文件夹名
    newname=path.with_suffix('.txt')
    # print(newname.name)
    file=open(target_dir/newname.name,'w')
    for i in range(len(back)):
        l=' '
        if (i+1)%5==0:
            l='\n'
        file.writelines(str(back[i])+l)#完成了，现在进行批量操作修改



print('转换完成了,大概\n')



#完成

下面是我用YOLOv5 跑的结果（需要具体步骤私信）电脑3060

评价指标

数据增强

 

 

 

训练过程

 

 结果