本文介绍如何使用Python脚本批量删除LabelImg创建的.xml文件中不需要的标签类，只保留指定类别（如people和cat），并转换标签值为小写，适用于调整训练数据集。
概述
自己在用labelImg打好标签后，想只用其中几类训练，不想训练全部类别，又不想重新打标生成.xml文件，因此想到这个办法：直接在.xml文件中删除原有的不需要的标签类及其属性。
打标时标签名出现了大小写(工程量大时可能会手滑)，程序中有改写标签值为小写的过程，因为我做py-faster-rcnn 训练时，标签必须全部为小写。
以如下的.xml文件为例，我故意把标签增加了大写
test.jpg
C:\Users\yasin\Desktop\test
Unknown
400
300
3
0
People
Unspecified
0
0
80
69
144
89
CAT
Unspecified
0
0
40
69
143
16
dog
Unspecified
0
0
96
82
176
87
具体实现
假如我们只想保留图片上的people和cat类，其他都删除，代码如下：
from xml.etree.ElementTree import ElementTree
from os import walk, path
def read_xml(in_path):
tree = ElementTree()
最低0.47元/天 解锁文章
扫一扫
3656
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
