python实用脚本（二）：删除xml标签下的指定种别

介绍

在目标检测中，有些时候会碰到标注好的种别不想要了的情况，这时我们可以运行下面的代码来批量删除不需要的种别节流时间。
代码实现：

2. import argparse
4. import xml.etree.ElementTree as ET
5. import os
7. classes = ['thin_smoke']
10. def GetImgNameByEveryDir(file_dir, videoProperty):
11.     FileNameWithPath, FileName, FileDir = [], [], []
12.     for root, dirs, files in os.walk(file_dir):
13.         for file in files:
14.             if os.path.splitext(file)[1] in videoProperty:
15.                 FileNameWithPath.append(os.path.join(root, file))  # 保存图片路径
16.                 FileName.append(file)  # 保存图片名称
17.                 FileDir.append(root[len(file_dir):])  # 保存图片所在文件夹
18.     return FileName, FileNameWithPath, FileDir
21. def GetBoxInfo(xmlfile):
22.     try:
23.         tree = ET.parse(xmlfile)
24.         root = tree.getroot()
25.         size = root.find('size')
26.         w = int(size.find('width').text)
27.         h = int(size.find('height').text)
28.     except:
29.         return False, 0
30.     else:
31.         tree = ET.parse(xmlfile)
32.         root = tree.getroot()
33.         size = root.find('size')
34.         w = int(size.find('width').text)
35.         h = int(size.find('height').text)
37.     for obj in root.findall('object'):
38.         print(xmlfile, obj.find('name').text, obj)
39.         if obj.find('name').text == 'fulll':
40.             obj.find('name').text = "full"
42.     print('-' * 66)
43.     tree.write(xmlfile)
45.     return 1
48. def Process(ProcessDir):
49.     xmlDirs = ProcessDir  # + 'Labels/'
50.     FileName1, FileNameWithPath1, FileDir1 = GetImgNameByEveryDir(xmlDirs, '.xml')
52.     for k in range(len(FileName1)):
53.         annfile = xmlDirs + FileName1[k][:-4] + '.xml'
54.         result = GetBoxInfo(annfile)
57. if __name__ == '__main__':
58.     parser = argparse.ArgumentParser()
59.     parser.add_argument('--ProcessDir', type=str, default='/home/build/smoke_ori/VOCdevkit/VOC2007/Annotations/')
60.     args = parser.parse_args()
61.     ProcessDir = args.ProcessDir
62.     Process(ProcessDir)

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。