binary_mask_from_json/binary_mask_from_json.py at main · maftouni/binary_mask_from_json · 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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import json
import os
import numpy as np
import PIL.Image
import cv2
import matplotlib.pyplot as plt


with open("annotation.json", "r") as read_file:
    data = json.load(read_file)

all_file_names=list(data.keys())

Files_in_directory = []
for root, dirs, files in os.walk("sample_frames"):
    for filename in files:
        Files_in_directory.append(filename)

for j in range(len(all_file_names)):
    image_name=data[all_file_names[j]]['filename']
    if image_name in Files_in_directory:
         img = np.asarray(PIL.Image.open('sample_frames/'+image_name))
    else:
        continue

    if data[all_file_names[j]]['regions'] != {}:
        #cv2.imwrite('images/%05.0f' % j +'.jpg',img)
        print(j)
        try:
             shape1_x=data[all_file_names[j]]['regions']['0']['shape_attributes']['all_points_x']
             shape1_y=data[all_file_names[j]]['regions']['0']['shape_attributes']['all_points_y']
        except :
             shape1_x=data[all_file_names[j]]['regions'][0]['shape_attributes']['all_points_x']
             shape1_y=data[all_file_names[j]]['regions'][0]['shape_attributes']['all_points_y']

        fig = plt.figure()

        plt.imshow(img.astype(np.uint8))
        plt.scatter(shape1_x,shape1_y,zorder=2,color='red',marker = '.', s= 55)


        ab=np.stack((shape1_x, shape1_y), axis=1)

        img2=cv2.drawContours(img, [ab], -1, (255,255,255), -1)


        mask = np.zeros((img.shape[0],img.shape[1]))
        img3=cv2.drawContours(mask, [ab], -1, 255, -1)

        cv2.imwrite('binary_masks/%05.0f' % j +'.png',mask.astype(np.uint8))