转载：删除xml文件中不需要的节点及其子信息（VOC格式、python语言）

最新推荐文章于 2021-09-17 11:32:21 发布

小祥子ovo

最新推荐文章于 2021-09-17 11:32:21 发布

阅读量413

点赞数 1

分类专栏：数据集

原文链接：https://blog.csdn.net/qq_43211132/article/details/109507781?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522162605705816780262538229%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=162605705816780262538229&biz_i

版权

python xml

数据集专栏收录该内容

2 篇文章 0 订阅

订阅专栏

原文章：删除xml文件中不需要的节点及其子信息（VOC格式、python语言）

import xml.etree.cElementTree as ET
import os

path_root = [r'C:\Users\18125\Desktop\remove&merge\Annotations_insulator']  #自己的xml路径

CLASSES = [ "insulator"]  #要保留的类别
for anno_path in path_root:
    xml_list = os.listdir(anno_path)
    for axml in xml_list:
        path_xml = os.path.join(anno_path, axml)
        print(path_xml)
        tree = ET.parse(path_xml)
        root = tree.getroot()

        for child in root.findall('object'):
            name = child.find('name').text
            if not name in CLASSES:
                root.remove(child)
        print(axml)
        tree.write(os.path.join(r'C:\Users\18125\Desktop\remove&merge\new_annotations', axml))  #处理结束后保存的路径