Python统计labelme的标签个数

说明

通过Labelme进行文件标注后，可通过下面的代码统计JSON文件中各类标签的名称和总数

在打标签的时候，我将生成的JSON文件和原图存放在了一个文件夹，因此在代码中要先遍历文件夹中的每一个文件，之后读取JSON文件 。代码输出标签的总类别数，并且显示自己命名的标签名称，同时按照英文字母的顺序对标签排序。代码可做参考，修改path路径即可。

参考代码

import json
import os


def count_labels_in_folder(path):
    label_counts = {}  # 存储标签名称和计数的字典


    for file_name in os.listdir(path):
        if file_name.endswith('.json'):
            json_file_path = os.path.join(path, file_name)
            with open(json_file_path, 'r') as json_file:
                data = json.load(json_file)
            for shape in data['shapes']:
                label_str = shape['label']
                if label_str in label_counts:
                    label_counts[label_str] += 1
                else:
                    label_counts[label_str] = 1
    sorted_labels = sorted(label_counts.keys())
    return label_counts, sorted_labels


if __name__ == "__main__":
    path = "Own the path"
    label_counts, sorted_labels = count_labels_in_folder(path)

    # 计算用于对齐输出的字段宽度，将输出的Count对齐
    max_label_length = max(len(label) for label in label_counts.keys())
    max_count_length = max(len(str(count)) for count in label_counts.values())

    total_categories = len(label_counts)
    print(f"Total Categories: {total_categories}\n")


    for label in sorted_labels:
        count = label_counts[label]
        label_padding = max_label_length - len(label)
        count_padding = max_count_length - len(str(count))
        print(f"Label: {label}{label_padding * ' '}     Count: {' ' * count_padding}{count}")

运行结果

 如有错误，欢迎指正！