voc xml标签批量修改,更改、删除指定类别

适用场景:

例如:适用于在数据标注时有部分标签名错误或需要更改、删除某一指定类别


修改指定类别名称

import os.path
import xml.dom.minidom

# xml文件存放路径
path = 'C:/Users/soli/Desktop/1'
# 定义需替换的类名['原始类别名称','新名称']
name = ['a', 'b']

files = os.listdir(path)  # 返回文件夹中的文件名列表
for xmlFile in files:
    dom = xml.dom.minidom.parse(path + '\\' + xmlFile)
    root = dom.documentElement
    newfilename = root.getElementsByTagName('name')
    for i, t in enumerate(newfilename):
        if t.firstChild.data == name[0]:
            newfilename[i].firstChild.data = name[1]
    with open(os.path.join(path, xmlFile), 'w') as fh:
        dom.writexml(fh)


删除指定类别

#  批量移除xml标注中的某一个类别标签
import xml.etree.cElementTree as ET
import os

# xml文件路径
xml_path = r'C:\Users\soli\Desktop\1'
xml_files = os.listdir(xml_path)
# 需要删除的类别名称
CLASSES = ["ssss"]

for axml in xml_files:
    path_xml = os.path.join(xml_path, axml)
    tree = ET.parse(path_xml)
    root = tree.getroot()
    for child in root.findall('object'):
        name = child.find('name').text
        if name in CLASSES:
            root.remove(child)
    tree.write(os.path.join(xml_path, axml))

  • 12
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

soli-lu

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值