（七）每个类别计数，基于yolo格式标签

原创已于 2024-04-21 19:30:04 修改 · 470 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言

于 2023-10-07 18:21:23 首次发布

YOLO训练tips 专栏收录该内容

7 篇文章

订阅专栏

文章讲述了如何使用Python的os库操作文件，统计在BDD100K数据集中val标签文件夹中各类别标签的出现次数，用于对象检测任务中的类别分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

# 标签文件夹路径
annotation_folder = "D:/PycharmProjects/yolov5-master/bdd100knightyolo/labels/val"

# 统计各类别标签数量的字典
label_counts = {}

# 获取标签文件夹中所有标签文件的路径
annotation_files = os.listdir(annotation_folder)

# 遍历每个标签文件
for annotation_file in annotation_files:
    # 标签文件路径
    annotation_path = os.path.join(annotation_folder, annotation_file)

    # 读取标签数据
    with open(annotation_path, "r") as file:
        annotations = file.readlines()

    # 统计每个标签出现的次数
    for annotation in annotations:
        label = annotation.split()[0]
        if label in label_counts:
            label_counts[label] += 1
        else:
            label_counts[label] = 1

# 打印各类别标签数量
for label, count in label_counts.items():
    print(f"{label}: {count}")