| | import os
|
| | import io
|
| | import json
|
| | import numpy as np
|
| | from pycococreatortools import pycococreatortools
|
| | from PIL import Image
|
| | import base64
|
| | import cv2
|
| | from tqdm import tqdm
|
| | def img_tobyte(img_pil):
|
| |
|
| | ENCODING = 'utf-8'
|
| | img_byte = io.BytesIO()
|
| | img_pil.save(img_byte, format='PNG')
|
| | binary_str2 = img_byte.getvalue()
|
| | imageData = base64.b64encode(binary_str2)
|
| | base64_string = imageData.decode(ENCODING)
|
| | return base64_string
|
| |
|
| | def mask2json(ROOT_DIR):
|
| | Image_DIR = os.path.join(ROOT_DIR, "pngs")
|
| | Label_DIR = os.path.join(ROOT_DIR, "pre")
|
| |
|
| | Label_files = os.listdir(Label_DIR)
|
| |
|
| | class_names = ['_background_', 'panicle']
|
| | for Label_filename in tqdm(Label_files):
|
| |
|
| | Json_output = {
|
| | "version": "3.16.7",
|
| | "flags": {},
|
| | "fillColor": [255, 0, 0, 128],
|
| | "lineColor": [0, 255, 0, 128],
|
| | "imagePath": {},
|
| | "shapes": [],
|
| | "imageData": {}}
|
| | name = Label_filename.split('.', 3)[0]
|
| | name1 = name + '.png'
|
| | Json_output["imagePath"] = name1
|
| |
|
| | image = Image.open(Image_DIR + '/' + name1)
|
| | imageData = img_tobyte(image)
|
| | Json_output["imageData"] = imageData
|
| | binary_mask = np.asarray(np.array(Image.open(Label_DIR + '/' + Label_filename))).astype(np.uint8)
|
| | mask_image = cv2.imread(Label_DIR + '/' + Label_filename, cv2.IMREAD_GRAYSCALE)
|
| | temp_mask = np.asarray((mask_image != 0), dtype=np.uint8)
|
| |
|
| | segmentation = pycococreatortools.binary_mask_to_polygon(temp_mask, tolerance=2)
|
| | for item in segmentation:
|
| | if len(item) > 10:
|
| | list1 = []
|
| | for j in range(0, len(item), 2):
|
| | list1.append([item[j], item[j + 1]])
|
| |
|
| | label = class_names[1]
|
| | seg_info = {'points': list1, "fill_color": None, "line_color": None, "label": label,
|
| | "shape_type": "polygon", "flags": {}}
|
| | Json_output["shapes"].append(seg_info)
|
| |
|
| | Json_output["imageHeight"] = binary_mask.shape[0]
|
| | Json_output["imageWidth"] = binary_mask.shape[1]
|
| | json_path = os.path.join(ROOT_DIR,'json')
|
| | if not os.path.exists(json_path):
|
| | os.makedirs(json_path)
|
| | full_path = os.path.join(json_path, '{}.json'.format(name))
|
| | with open(full_path, 'w') as output_json_file:
|
| | json.dump(Json_output, output_json_file)
|
| |
|
| |
|
| | if __name__ == '__main__':
|
| |
|
| | ROOT_DIR = ''
|
| | mask2json(ROOT_DIR)
|
| |
|
| |
|
