Private-code/json_data.py at master · sun9545/Private-code · GitHub

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
29
30
31
32
33
34
import os
import cv2
import numpy as np

# 将json文件label转换为到data文件夹
n = 498  # n为总共标注的图片数
for i in range(n):
    path = '/home/sjq/img/data/%d_json'%(i)
    if not os.path.exists(path):
        os.makedirs(path)
    os.system('labelme_json_to_dataset /home/sjq/img/output/%d.json -o /home/sjq/img/data/%d_json' % (i, i))
# dst_w=512
# dst_h=512
# dst_shape=(dst_w,dst_h,3)
train_image = '/home/sjq/img/train_image/'
if not os.path.exists(train_image):
    os.makedirs(train_image)
train_label = '/home/sjq/img/train_label/'
if not os.path.exists(train_label):
    os.makedirs(train_label)

for i in range(n):
    print(i)
    img = cv2.imread('/home/sjq/img/data/%d_json/img.png' % i)
    label = cv2.imread('/home/sjq/img/data/%d_json/label.png' % i)
    print(img.shape)
    label = label / np.max(label[:, :, 2]) * 255
    label[:, :, 0] = label[:, :, 1] = label[:, :, 2]
    print(np.max(label[:, :, 2]))
    # cv2.imshow('l',label)
    # cv2.waitKey(0)
    print(set(label.ravel()))
    cv2.imwrite(train_image + '%d.jpg' % i, img)
    cv2.imwrite(train_label + '%d.jpg' % i, label)