###  计算xml中每个类别标签的数量  ###

import os
import shutil
import xml.etree.ElementTree as ET

def caculate_label_numble(file_path):
    label_numble_info = {}
    for file in os.listdir(file_path):
        if file.endswith(".xml"):
            xml_path = os.path.join(file_path, file)
            tree = ET.parse(xml_path)
            root = tree.getroot()

            for element in root.findall('object'):
                for element1 in element.findall('name'):
                    if element1.text not in label_numble_info:
                        # if element1.text=="6935284412918":#查看异常分类对应的xml
                        #     print(file)
                        label_numble_info[element1.text] = 1
                    else:
                        label_numble_info[element1.text] += 1
    return label_numble_info

if __name__=="__main__":

    file_path = r"D:\PycharmProjects\ieemoo-ai-pack\paper_data\Annotations"
    label_numble_info = caculate_label_numble(file_path)
    for key, value in label_numble_info.items():
        print('{}:{}'.format(key, value))