YOLOv8-OBB 旋转模型如何训练自己的模型(记录)

cls_list=[‘分类名称’]
def edit_xml(xml_file, dotaxml_file):
“”"
修改xml文件
:param xml_file:xml文件的路径
:return:
“”"
tree = ET.parse(xml_file)
objs = tree.findall(‘object’)
for ix, obj in enumerate(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”)
# obj_type = obj.find(‘bndbox’)
# type = obj_type.text
# print(xml_file)

    if (obj.find('robndbox') == None):
        obj_bnd = obj.find('bndbox')
        obj_xmin = obj_bnd.find('xmin')
        obj_ymin = obj_bnd.find('ymin')
        obj_xmax = obj_bnd.find('xmax')
        obj_ymax = obj_bnd.find('ymax')
        #以防有负值坐标
        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'  # 修改节点名