将目标检测的json文件可视化

在训练目标检测网络时，groundtruth常以json格式存储，以下代码可实现json文件可视化：

json文件内格式：
{“rotate”: 0, “objects”: [{“rect”: [210, 45, 231, 30], “label”: “Cat”}]}

#coding:utf-8
import os
import json
import cv2
import sys
reload(sys)
sys.setdefaultencoding('utf8')

def json_visualization(json_path,img_path,name,marksave_dir):
    marksave_path = os.path.join(marksave_dir, name + '.jpg')
    print(marksave_path)
    print(img_path)
    i = 0
    with open(json_path, 'r') as json_file:
        target = []
        img_ori = cv2.imread(img_path)
        h_o, w_o, c_o = img_ori.shape
        img_mark = img_ori.copy()
        #print(json_file)
        for line in json_file:
            data = json.loads(line)
            objs = data['objects']
            classes = ['Cat']
            for obj in objs:
                label = obj['label']
                # print(label)
                if label not in classes:
                    continue
                p = obj['rect']
                xmin = int(p[0])
                ymin = int(p[1])
                width = int(p[2])
                height = int(p[3])
                xmax = xmin + width
                ymax = ymin +height

                if xmin<0 or ymin<0 or xmin+width>w_o or ymin+height>h_o:
                    print('Invalid point~')
                    print(img_path)
                    continue

                target.append([label,xmin,ymin,xmin+width,ymin+height])
                color = (0,0,255) if label=='Chn' else (0,255,255)
                cv2.rectangle(img_mark, (xmin , ymin), (xmax, ymax), color, 2)

            marksave_path = os.path.join(marksave_dir, name+'.jpg')
            cv2.imwrite(marksave_path, img_mark)

def image_json_visualization(img_dir,json_dir,save_dir):
    '''
    :param img_dir: 图片所在文件夹
    :param json_dir: json文件所在文件夹
    :param save_dir: 可视化后图片保存路径
    :return:
    '''
    marksave_dir=os.path.join(save_dir,'mark1')
    if not os.path.exists(marksave_dir):
        os.makedirs(marksave_dir)
    for item in os.listdir(json_dir):
        json_path = os.path.join(json_dir, item)
        name = item[:-5]
        img_path = os.path.join(img_dir,str(name)+'.png')
        json_visualization(json_path,img_path,name,marksave_dir)
if __name__ == '__main__':
    img_dir = r'图片所在文件夹'
    json_dir = r'json文件所在文件夹'
    save_dir = r'可视化后图片保存路径'
    image_json_visualization(img_dir,json_dir,save_dir)