用python写一段代码用来统计voc数据集标签中的目标真实框的高度和宽度的分布

可以使用以下代码来统计voc数据集标签中的目标真实框的高度和宽度的分布：

import xml.etree.ElementTree as ET
import os
import numpy as np
import matplotlib.pyplot as plt

# VOC数据集标签所在的文件夹路径
anno_dir = 'path/to/annotations'

# 统计目标真实框的高度和宽度
heights = []
widths = []
for filename in os.listdir(anno_dir):
    if filename.endswith('.xml'):
        tree = ET.parse(os.path.join(anno_dir, filename))
        root = tree.getroot()
        for obj in root.findall('object'):
            bbox = obj.find('bndbox')
            xmin = int(bbox.find('xmin').text)
            ymin = int(bbox.find('ymin').text)
            xmax = int(bbox.find('xmax').text)
            ymax = int(bbox.find('ymax').text)
            width = xmax - xmin
            height = ymax - ymin
            widths.append(width)
            heights.append(height)

# 绘制高度和宽度的分布直方图
fig, axs = plt.subplots(1, 2, figsize=(10, 5))
axs[0].hist(heights, bins=50)
axs[0].set_title('Height Distribution')
axs[0].set_xlabel('Height')
axs[0].set_ylabel('Count')
axs[1].hist(widths, bins=50)
axs[1].set_title('Width Distribution')
axs[1].set_xlabel('Width')
axs[1].set_ylabel('Count')
plt.show()

这段代码会遍历VOC数据集标签文件夹中的所有XML文件，提取每个目标真实框的高度和宽度，并将它们存储在两个列表中。然后，使用Matplotlib库绘制高度和宽度的分布直方图。