Python统计labelme标注Json文件的标签数

前言

• 本文是个人使用PythonPython处理文件的电子笔记，由于水平有限，难免出现错漏，敬请批评改正。
• 更多精彩内容，可点击进入我的个人主页查看

前提条件

• 熟悉Python

相关介绍

• Python是一种跨平台的计算机程序设计语言。是一个高层次的结合了解释性、编译性、互动性和面向对象的脚本语言。最初被设计用于编写自动化脚本(shell)，随着版本的不断更新和语言新功能的添加，越多被用于独立的、大型项目的开发。
• Python OS模块是负责程序与操作系统的交互，提供了访问操作系统底层的接口和非常丰富的方法用来处理文件和目录。

实验环境

• Python 3.x （面向对象的高级语言）

Python统计labelme标注单个Json文件的标签数

代码实现

import os
import cv2
import json


def count_json_label(in_json_path):
    label_num = {}
    with open(in_json_path, "r", encoding='utf-8') as f:
        # json.load数据到变量json_data
        json_data = json.load(f)
    for i in json_data['shapes']:
        if i['label'] in label_num.keys():
            label_num[i['label']] = label_num[i['label']] + 1
        else:
            label_num[i['label']] = 1
    print(label_num)

if __name__=="__main__":
    in_dir_name = 'origin/' # 图片和json所在文件夹
    out_dir_name = 'output/'
    all_name_list = os.listdir(in_dir_name) # 获取文件夹内所有文件名列表

    # 获取所有json文件，并修改
    json_name_list = [i for i in all_name_list if i.endswith('.json')] # 图片路径列表
    # print(json_name_list)
    for json_name in json_name_list: # 遍历json文件并修改
        in_json_path = in_dir_name + json_name
        count_json_label(in_json_path) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

{'1': 4, '0': 3, '6': 1}
{'1': 5, '6': 2, '0': 3}
1
2

Python统计labelme标注多个Json文件的标签数

import os
import cv2
import json


def count_json_label(in_json_path,label_num):
    with open(in_json_path, "r", encoding='utf-8') as f:
        # json.load数据到变量json_data
        json_data = json.load(f)
    for i in json_data['shapes']:
        if i['label'] in label_num.keys():
            label_num[i['label']] = label_num[i['label']] + 1
        else:
            label_num[i['label']] = 1

if __name__=="__main__":
    in_dir_name = 'origin/' # 图片和json所在文件夹
    out_dir_name = 'output/'
    all_name_list = os.listdir(in_dir_name) # 获取文件夹内所有文件名列表

    # 获取所有json文件，并修改
    json_name_list = [i for i in all_name_list if i.endswith('.json')] # 图片路径列表
    # print(json_name_list)
    label_num = {}
    for json_name in json_name_list: # 遍历json文件并修改
        in_json_path = in_dir_name + json_name
        count_json_label(in_json_path,label_num)
    print(label_num)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

{'1': 9, '0': 6, '6': 3}
1

更多精彩内容，可点击进入Python日常小操作专栏或我的个人主页查看