Small-code: 计算xml格式数据集中每一类的数量

该代码用于计算VOC格式数据集中每一类标注框的数量：

# -*- coding:utf-8 -*-
import os
import xml.etree.ElementTree as ET
import numpy as np

# np.set_printoptions(suppress=True, threshold=np.nan)
import matplotlib
from PIL import Image


def parse_obj(xml_path, filename):
    tree = ET.parse(xml_path + filename)
    objects = []
    for obj in tree.findall('object'):
        obj_struct = {}
        obj_struct['name'] = obj.find('name').text
        objects.append(obj_struct)
    return objects


def read_image(image_path, filename):
    im = Image.open(image_path + filename)
    W = im.size[0]
    H = im.size[1]
    area = W * H
    im_info = [W, H, area]
    return im_info


if __name__ == '__main__':
    # xml_path = 'E:\\dataset\\solar_cell_EL_image\\PVELAD\\EL2021\\test\\Annotations\\'
    xml_path =  'E:\\dataset\\solar_cell_EL_