Python筛选出含有指定label的Json文件

前言

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

前提条件

• 熟悉Python

相关介绍

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

实验环境

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

Python筛选出含有指定label的Json文件

import os
import cv2
import json
import shutil
import numpy as np

def select_img_json(img_path,in_json_path,out_img_path,out_json_path,label_name):
    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']==label_name:
            shutil.copy(img_path, out_img_path)
            shutil.copy(in_json_path, out_json_path)
            continue

if __name__ == "__main__":
    images_dir = 'trainset/images/' # 图片文件夹
    jsons_dir = 'trainset/jsons/' # json文件夹
    output_dir = 'trainset/output/' # 筛选文件所在的文件夹

    if not os.path.exists(output_dir):
        os.mkdir(output_dir)

    images_dir_list = os.listdir(images_dir)
    images_list = [i for i in images_dir_list if i.endswith(
        '.jpg') or i.endswith('.png') or i.endswith('.bmp')]
    # print(images_list)

    jsons_dir_list = os.listdir(jsons_dir)
    jsons_list = [i for i in jsons_dir_list if i.endswith('.json')]
    # print(jsons_list)
   
    label_name = input("请输入要筛选的label_name:")
    for image in images_list:
        try:
            img_path = images_dir + image
            in_json_path = jsons_dir + image[:-4] + '.json'
            out_img_path = output_dir + image
            out_json_path = output_dir + image[:-4] + '.json'
            select_img_json(img_path,in_json_path,out_img_path,out_json_path,label_name)
        except Exception as e:
            print(e)

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