• remove:删除结点
  • append:添加节点
import xml.etree.ElementTree as ET


# 删除某一固定类别的结点
def removecls(xml_path, cls):
    tree = ET.parse(xml_path)  # 保存xml文件是tree
    root = tree.getroot()  # 修改xml问价是对root操作

    # 通常用find来获得某一结点的地址,然后再继续访问一些元素值
    # 通过findall或者iterfind,都可以直接获取到所有object的结点地址
    objects = root.findall('object')
    for obj in objects:
        cls = obj.find('name').text
        if cls == 'loc':
            root.remove(obj)

    tree.write('caogao.xml')    # 保存xml


# 加入一个object节点
def appendobj(xml_path, name, xmin, ymin, xmax, ymax):
    tree = ET.parse(xml_path)
    root = tree.getroot()

    new_obj = ET.Element('object')
    new_obj.tail = '\n'
    root.append(new_obj)

    new_obj_name = ET.Element('name')
    new_obj_name.text = name
    new_obj.append(new_obj_name)

    new_obj_box = ET.Element('bndbox')
    new_obj.append(new_obj_box)

    new_obj_box_xmin = ET.Element('xmin')
    new_obj_box_xmin.text = str(xmin)
    new_obj_box.append(new_obj_box_xmin)

    new_obj_box_ymin = ET.Element('ymin')
    new_obj_box_ymin.text = str(ymin)
    new_obj_box.append(new_obj_box_ymin)

    new_obj_box_xmax = ET.Element('xmax')
    new_obj_box_xmax.text = str(xmax)
    new_obj_box.append(new_obj_box_xmax)

    new_obj_box_ymax = ET.Element('ymax')
    new_obj_box_ymax.text = str(ymax)
    new_obj_box.append(new_obj_box_ymax)

    tree.write('caogao2.xml')   # 保存xml


if __name__ == '__main__':
    xml_path = r'G:\pycharmprojects\ScriptCodes_lwd\linepaste\lines\xmlloc\1a.xml'

    # 1.删除某一结点
    removecls(xml_path, '')

    # 2.添加一个结点
    # appendobj(xml_path, 'loc', 200, 300, 300, 400)
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐