20220418月曜日

import json
import glob
import shutil
import os

       
SRC_PATH = --------
DEST_PATH = --------

TARGET = 'SM'

def read_json():
    list_of_files = glob.glob(SRC_PATH + "*.json")
#     list_of_files = glob.glob("D:/files_lyw/test_files/*.json")
    for file_name in list_of_files:
        fin = open(file_name, "rt", encoding='UTF8')
        json_data = json.load(fin)
        count_of_labels = len(json_data["shapes"])
        count_of_target_object = 0
        if count_of_labels > 0:
            for index in range(count_of_labels):
                label_name = str(json_data["shapes"][index]["label"]).strip().split('_')
                object_name = label_name[0]
                if object_name == TARGET:
                    count_of_target_object += 1
                    break
        fin.close()
        if count_of_target_object >= 1:
            file_name_without_ext, ext = os.path.splitext(file_name)
            base_file_name = os.path.basename(file_name_without_ext)
            try:
                shutil.move(file_name, DEST_PATH + base_file_name +".json")
                shutil.move(file_name_without_ext + '.jpg', DEST_PATH + base_file_name + '.jpg')
            except:
                print(file_name + ' 경로 오류')
        
read_json()