獲取voc數據集標籤所有的類別數

import xml.dom.minidom as xmldom
import os

#voc數據集獲取所有標籤的所有類別數"
annotation_path=".\VOC2007\Annotations"

annotation_names=[os.path.join(annotation_path,i) for i in os.listdir(annotation_path)]

labels = list()
for names in annotation_names:
    xmlfilepath = names
    domobj = xmldom.parse(xmlfilepath)
    # 得到元素對象
    elementobj = domobj.documentElement
    #獲得子標籤
    subElementObj = elementobj.getElementsByTagName("object")
    for s in subElementObj:
        label=s.getElementsByTagName("name")[0].firstChild.data
        #print(label)
        if label not in labels:
            labels.append(label)
print(labels)