功能:能够删除xml中不需要的类别标签,和替换制定的类别名称。
使用方法:
1)修改origin_ann_dir:原始xml文件夹路径
2)修改new_ann_dir: 保存新xml的文件夹路径
3)需要修改和删除的类别名称、
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import os
import xml.etree.ElementTree as ET
origin_ann_dir = '/data_1/xml/'# 设置原始标签路径为 Annos
new_ann_dir = '/data_1/new_xml/'# 设置新标签路径 Annotations
for dirpaths, dirnames, filenames in os.walk(origin_ann_dir): # os.walk游走遍历目录名
for filename in filenames:
print("process...")
if os.path.isfile(r'%s%s' %(origin_ann_dir, filename)): # 获取原始xml文件绝对路径,isfile()检测是否为文件 isdir检测是否为目录
origin_ann_path = os.path.join(r'%s%s' %(origin_ann_dir, filename)) # 如果是,获取绝对路径(重复代码)
new_ann_path = os.path.join(r'%s%s' %(new_ann_dir, filename))
tree = ET.parse(origin_ann_path) # ET是一个xml文件解析库,ET.parse()打开xml文件。parse--"解析"
root = tree.getroot() # 获取根节点
for object in root.findall('object'): # 找到根节点下所有“object”节点
name = str(object.find('name').text) # 找到object节点下name子节点的值(字符串)
# 如果name等于str,则删除该节点
if (name in ["car_head"]):
root.remove(object)
# 如果name等于str,则修改name
if(name in ["other_light"]):
object.find('name').text = "person"
# 检查是否存在labelmap中没有的类别
for object in root.findall('object'):
name = str(object.find('name').text)
if not (name in ["chepai","chedeng","chebiao"]):
print(filename + "------------->label is error--->" + name)
tree.write(new_ann_path)#tree为文件,write写入新的文件中。
功能二:直接替换xml中的某个文本信息
#-*- coding:utf-8 -*-
#将a替换成b
import os
xmldir='/data_1/data/标注数据/车牌/result/车牌和螺丝/convert_xml'
savedir='/data_1/data/标注数据/车牌/result/车牌和螺丝/new_xml'
xmllist=os.listdir(xmldir)
for xml in xmllist:
if '.xml' in xml:
fo=open(savedir+'/'+'{}'.format(xml),'w')#不要new前缀可以在这改
print('{}'.format(xml))
fi=open(xmldir+'/'+'{}'.format(xml),'r')
content=fi.readlines()
for line in content:
line=line.replace('>','>')#例:将a替换为b
line=line.replace(' </annotation>','</annotation>')#例:将a替换为b
fo.write(line)
fo.close()
print('替换成功')