批量删除VOC的XML中的某些节点

最新推荐文章于 2024-01-11 09:16:12 发布

P0ny

最新推荐文章于 2024-01-11 09:16:12 发布

阅读量1.7k

点赞数 5

分类专栏： python 文章标签：删除节点 xml python voc

本文链接：https://blog.csdn.net/qq_36735489/article/details/80732346

版权

python 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

import xml.etree.cElementTree as ET
import os
path_root = ['E:\data-VOC0712\VOC2007\Annotations',
             'E:\data-VOC0712\VOC2012\Annotations']

CLASSES = [
           "bottle",
           "cat", "chair",  "diningtable",
           "dog", "motorbike", "person",
           "pottedplant","sofa",
           "tvmonitor"]
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)
        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)

        tree.write(os.path.join('E:\data-myVOC0712\Annotations', axml))