Yolov8旋转目标检测自建数据集训练（XML转txt）

 在参照CSDN中：《【保姆级教程】YOLOv8_OBB旋转目标检测：训练自己的数据集》博客中XML转txt报错，有问题。更改以后的代码如下：

import os
import xml.etree.ElementTree as ET
import math

cls_list = ['up', 'down']  # 修改为自己的标签

def edit_xml(xml_file, dotaxml_file):
    """
    修改xml文件
    :param xml_file:xml文件的路径
    :return:
    """
    tree = ET.parse(xml_file)
    objs = tree.findall('object')
    for obj in objs:
        x0 = ET.Element("x0")  # 创建节点
        y0 = ET.Element("y0")
        x1 = ET.Element("x1")
        y1 = ET.Element("y1")
        x2 = ET.Element("x2")
        y2 = ET.Element("y2")
        x3 = ET.Element("x3")
        y3 = ET.Element("y3")

        if obj.find('robndbox') is None:
            obj_bnd = obj.find('bndbox')
            if obj_bnd is None:
                continue  # 如果没有找到bndbox节点，则跳过这个object

            obj_xmin = obj_bnd.find('xmin')
            obj_ymin = obj_bnd.find('ymin')
            obj_xmax = obj_bnd.find('xmax')
            obj_ymax = obj_bnd.find('ymax')

            if obj_xmin is None or obj_ymin is None or obj_xmax is None or obj_ymax is None:
                continue  # 如果任意一个坐标缺失，则跳过这个object

            # 以防有负值坐标
            xmin = max(float(obj_xmin.text), 0)
            ymin = max(float(obj_ymin.text), 0)
            xmax = max(float(obj_xmax.text), 0)
            ymax = max(float(obj_ymax.text), 0)

            obj_bnd.remove(obj_xmin)  # 删除节点
            obj_bnd.remove(obj_ymin)
            obj_bnd.remove(obj_xmax)
            obj_bnd.remove(obj_ymax)

            x0.text = str(xmin)
            y0.text = str(ymax)
            x1.text = str(xmax)
            y1.text = str(ymax)
            x2.text = str(xmax)
            y2.text = str(ymin)
            x3.text = str(xmin)
            y3.text = str(ymin)
        else:
            obj_bnd = obj.find('robndbox')
            obj_bnd.tag = 'bndbox'  # 修改节点名
            obj_cx = obj_bnd.find('cx')
            obj_cy = obj_bnd.find('cy')
            obj_w = obj_bnd.find('w')
            obj_h = obj_bnd.find('h')
            obj_angle = obj_bnd.find('angle')

            if obj_cx is None or obj_cy is None or obj_w is None or obj_h is None or obj_angle is None:
                continue  # 如果任意一个旋转框参数缺失，则跳过这个object

            cx = float(obj_cx.text)
            cy = float(obj_cy.text)
            w = float(obj_w.text)
            h = float(obj_h.text)
            angle = float(obj_angle.text)

            obj_bnd.remove(obj_cx)  # 删除节点
            obj_bnd.remove(obj_cy)
            obj_bnd.remove(obj_w)
            obj_bnd.remove(obj_h)
            obj_bnd.remove(obj_angle)

            x0.text, y0.text = rotatePoint(cx, cy, cx - w / 2, cy - h / 2, -angle)
            x1.text, y1.text = rotatePoint(cx, cy, cx + w / 2, cy - h / 2, -angle)
            x2.text, y2.text = rotatePoint(cx, cy, cx + w / 2, cy + h / 2, -angle)
            x3.text, y3.text = rotatePoint(cx, cy, cx - w / 2, cy + h / 2, -angle)

        obj_bnd.append(x0)  # 新增节点
        obj_bnd.append(y0)
        obj_bnd.append(x1)
        obj_bnd.append(y1)
        obj_bnd.append(x2)
        obj_bnd.append(y2)
        obj_bnd.append(x3)
        obj_bnd.append(y3)

    tree.write(dotaxml_file, method='xml', encoding='utf-8')  # 更新xml文件

def rotatePoint(xc, yc, xp, yp, theta):
    xoff = xp - xc
    yoff = yp - yc
    cosTheta = math.cos(theta)
    sinTheta = math.sin(theta)
    pResx = cosTheta * xoff + sinTheta * yoff
    pResy = - sinTheta * xoff + cosTheta * yoff
    return str(int(xc + pResx)), str(int(yc + pResy))

def totxt(xml_path, out_path):
    files = os.listdir(xml_path)
    i = 0
    for file in files:
        tree = ET.parse(xml_path + os.sep + file)
        root = tree.getroot()
        name = file.split('.')[0]
        output = out_path + '/' + name + '.txt'
        with open(output, 'w') as file:
            objs = tree.findall('object')
            for obj in objs:
                cls = obj.find('name').text
                box = obj.find('bndbox')
                x0 = int(float(box.find('x0').text))
                y0 = int(float(box.find('y0').text))
                x1 = int(float(box.find('x1').text))
                y1 = int(float(box.find('y1').text))
                x2 = int(float(box.find('x2').text))
                y2 = int(float(box.find('y2').text))
                x3 = int(float(box.find('x3').text))
                y3 = int(float(box.find('y3').text))
                if x0 < 0:
                    x0 = 0
                if x1 < 0:
                    x1 = 0
                if x2 < 0:
                    x2 = 0
                if x3 < 0:
                    x3 = 0
                if y0 < 0:
                    y0 = 0
                if y1 < 0:
                    y1 = 0
                if y2 < 0:
                    y2 = 0
                if y3 < 0:
                    y3 = 0
                for cls_index, cls_name in enumerate(cls_list):
                    if cls == cls_name:
                        file.write("{} {} {} {} {} {} {} {} {} {}\n".format(x0, y0, x1, y1, x2, y2, x3, y3, cls, cls_index))
        i += 1
        print(f"Processed file {i}: {output}")

if __name__ == '__main__':
    roxml_path = '存放自己标注后的XML格式的路径'
    dotaxml_path = '存放带角度XML格式的路径'
    out_path = '存放转换后的txt格式的路径'
    filelist = os.listdir(roxml_path)
    for file in filelist:
        edit_xml(os.path.join(roxml_path, file), os.path.join(dotaxml_path, file))
    totxt(dotaxml_path, out_path)