将单张图像的json转化成图片

'''
将单张图像的json转化成图片
'''
import json
import re
import cv2
import numpy as np

def toRgb(tmp):
    opt = re.findall(r'(.{2})', tmp)  # 将字符串两两分割
    strs = ""  # 用以存放最后结果
    for i in range(0, len(opt)):  # for循环，遍历分割后的字符串列表
        strs += str(int(opt[i], 16)) + ","  # 将结果拼接成12，12，12格式
    aa = strs[0:-1]
    aa.split(',')[0]
    num = [int(aa.split(',')[0]), int(aa.split(',')[1]), int(aa.split(',')[2])]
    return num

if __name__ == "__main__":
    path = "./json_save_path/红河-烫金工艺-套准偏差_18_IPU3_534306_0.json"
    with open(path, 'r') as load_f:
        load_dict = json.load(load_f)
        print(len(load_dict))
        height = load_dict['Height']
        width = load_dict['Width']
        Lines = load_dict['Lines']
        Polygon=load_dict['Polygons']
        Bound = load_dict['FillRects']
        img = np.zeros((width, height), dtype=np.uint8)  # random.random()方法后面不能加数据类型
        bgr_img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
        if (Bound is not None) and (len(Bound) != 0): #矩形标注
            Bound = load_dict['FillRects']
            Bounds1 = Bound[0]['Bounds']

            a = [int(Bounds1.split(',')[0]), int(Bounds1.split(',')[1]), int(Bounds1.split(',')[2]),
                 int(Bounds1.split(',')[3])]
            color = Bound[0]['Color']
            tmp = color[1:]
            value = toRgb(tmp)
            value =(value[0],value[1],value[2])

            print("hello")
            triangle = np.array([[0, 0], [400, 0], [400, 400], [0, 400]])

            # rect = np.arange(8).reshape((4, 2))
            rect = np.array([[0 for col in range(2)] for row in range(4)])

            rect[0][0] = int(a[0])
            rect[0][1] = int(a[1])

            rect[1][0] = int(a[2])
            rect[1][1] = int(a[1])

            rect[2][0] = int(a[2])
            rect[2][1] = int(a[3])

            rect[3][0] = int(a[0])
            rect[3][1] = int(a[3])

            cv2.fillConvexPoly(bgr_img,rect,color=value)

        if len(Lines)==0 and len(Bound)==0:
        # if (Lines is None) and (Bound is None):
            bgr_img = bgr_img

        if len(Lines) !=0:  # 随机涂抹

            for i in range(len(Lines)):
                a = Lines[i]
                color = a['Color']
                tmp = color[1:]
                value = toRgb(tmp)
                points = a['Points']
                stroke = a['Stroke']
                arr = np.array([[0, 0]] * len(points))
                # 将points 转化为array
                for j in range(len(points)):
                    p = points[j]
                    pp1 = int(p.split(',')[0])
                    pp2 = int(p.split(',')[1])
                    arr[j, 0] = pp1
                    arr[j, 1] = pp2

                print('Line finished!')
                arr = arr.reshape(-1, 1, 2)
                cv2.polylines(bgr_img, [arr], False, (value[2], value[1], value[0]), thickness=stroke)

        cv2.imencode('.bmp', bgr_img)[1].tofile('图像_mask.bmp')

json中的内容：

{
	"FillRects" : null,
	"Height" : 200,
	"Indexs" : 
	[
		0
	],
	"Lines" : 
	[
		{
			"Color" : "#55AA00",
			"Index" : 0,
			"Points" : 
			[
				"79,106",
				"79,105",
				"80,105",
				"80,104",
				"81,104",
				"82,103",
				"83,103",
				"84,102",
				"85,102",
				"85,101",
				"86,101",
				"87,101",
				"87,100",
				"88,100",
				"89,100",
				"89,99",
				"90,99",
				"91,99",
				"92,99",
				"93,99",
				"94,99",
				"94,98",
				"95,98",
				"96,98",
				"97,98",
				"98,98",
				"98,99",
				"99,99",
				"100,99",
				"101,99",
				"102,99",
				"102,100",
				"103,100",
				"104,100",
				"104,101",
				"105,101",
				"105,102",
				"106,102",
				"107,103",
				"108,103",
				"108,104",
				"108,105",
				"109,105",
				"109,106",
				"109,107",
				"109,108",
				"109,109",
				"109,110",
				"109,111",
				"109,112",
				"108,112",
				"108,113",
				"108,114",
				"107,114",
				"107,115",
				"106,116",
				"106,117",
				"105,117",
				"104,117",
				"104,118",
				"103,118",
				"103,119",
				"102,120"
			],
			"Stroke" : 10
		}
	],
	"Operator" : "DL",
	"Polygons" : null,
	"Rubbers" : null,
	"Shapes" : 
	[
		1
	],
	"Width" : 200
}

生成的图像如下:

 

批量处理代码：

import json
import cv2
import numpy as np
import re

def toRgb(tmp):
    opt = re.findall(r'(.{2})', tmp)  # 将字符串两两分割
    strs = ""  # 用以存放最后结果
    for i in range(0, len(opt)):  # for循环，遍历分割后的字符串列表
        strs += str(int(opt[i], 16)) + ","  # 将结果拼接成12，12，12格式
    aa=strs[0:-1]
    aa.split(',')[0]
    num=[int(aa.split(',')[0]),int(aa.split(',')[1]),int(aa.split(',')[2])]
    return num

if __name__=="__main__":
    file_list_path='D:/CF_new/json_to_img/json_name.txt'
    path='D:/CF_new/json_to_img/saved_img/'
    with open(file_list_path, "r+") as flist:
        read_data = flist.read()
        # 对单张图像进行处理
        for eachline in read_data.split('\n'):
            file_name = eachline.split('/')[-1].split('.')[0]

            #读json文件
            with open(eachline, 'r') as load_f:
                load_dict = json.load(load_f)
                print(len(load_dict))
                height = load_dict['Height']
                width = load_dict['Width']
                Lines = load_dict['Lines']
                Polygon = load_dict['Polygons']
                Bound = load_dict['FillRects']
                img = np.zeros((width, height), dtype=np.uint8)  # random.random()方法后面不能加数据类型
                bgr_img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)

                if (Bound is not None) and (len(Bound)!=0):  # 矩形标注
                    Bound = load_dict['FillRects']
                    Bounds1 = Bound[0]['Bounds']

                    a = [int(Bounds1.split(',')[0]), int(Bounds1.split(',')[1]), int(Bounds1.split(',')[2]),
                         int(Bounds1.split(',')[3])]
                    color = Bound[0]['Color']
                    tmp = color[1:]
                    value = toRgb(tmp)
                    value = (value[2], value[1], value[0])

                    print("hello")
                    triangle = np.array([[0, 0], [400, 0], [400, 400], [0, 400]])

                    # rect = np.arange(8).reshape((4, 2))
                    rect = np.array([[0 for col in range(2)] for row in range(4)])

                    rect[0][0] = int(a[0])
                    rect[0][1] = int(a[1])

                    rect[1][0] = int(a[2])
                    rect[1][1] = int(a[1])

                    rect[2][0] = int(a[2])
                    rect[2][1] = int(a[3])

                    rect[3][0] = int(a[0])
                    rect[3][1] = int(a[3])

                    cv2.fillConvexPoly(bgr_img, rect, color=value)

                if ((Lines is None) and (Bound is None) ):
                    bgr_img=bgr_img

                if (Lines is not None):  # 随机涂抹
                    for i in range(len(Lines)):
                        a = Lines[i]
                        color = a['Color']
                        tmp = color[1:]
                        value = toRgb(tmp)
                        points = a['Points']
                        stroke = a['Stroke']
                        arr = np.array([[0, 0]] * len(points))
                        # 将points 转化为array
                        for j in range(len(points)):
                            p = points[j]
                            pp1 = int(p.split(',')[0])
                            pp2 = int(p.split(',')[1])
                            arr[j, 0] = pp1
                            arr[j, 1] = pp2

                        print('Line finished!')
                        arr = arr.reshape(-1, 1, 2)
                        cv2.polylines(bgr_img, [arr], False, (value[2], value[1], value[0]), thickness=stroke)



                # cv2.imwrite(path+file_name+'.bmp', bgr_img)
                cv2.imencode('.bmp', bgr_img)[1].tofile(path+file_name+'.bmp')