从一堆.xml标签文件中把object的类别提取出来
import xml.etree.ElementTree as ET
import os
import glob
file_dir = "E:\异物图片和标签\Annotations"
labelName = set([])
for files in os.walk(file_dir, topdown=False):
pppg=files[2]
for i in range(1449):
file_dir="E:/异物图片和标签/Annotations/"+pppg[i]
files = glob.glob(file_dir)
for i in range(len(files)):
AnotPath = files[i]
tree = ET.ElementTree(file=AnotPath)
root = tree.getroot()
ObjectSet = root.findall('object/name')
for Object in ObjectSet:
labelName.add(Object.text)
print(labelName)