统计标签中的类别数量

统计目标检测txt标签中各个目标类别的数量分布情况 

import os

dirpath = './2/'  # 存放转换后TXT文件的目录
output_file = '2.txt'  # 输出的统计结果文件名

class_counts = {}

# 遍历目标标签文件夹
for txt_file in os.listdir(dirpath):
    txt_path = os.path.join(dirpath, txt_file)
    # 读取每个标签文件
    with open(txt_path, 'r') as file:
        # 逐行读取标签文件内容
        for line in file:
            # 提取每行的目标类别
            label = line.strip().split()[0]
            # 更新类别计数
            class_counts[label] = class_counts.get(label, 0) + 1

# 将类别计数结果写入到文件中
with open(output_file, 'w') as outfile:
    for label, count in class_counts.items():
        outfile.write(f"{label}: {count}\n")

print('类别统计结果已写入到', output_file)