From 369b16483c669ea53cbbe144c872ce3167959049 Mon Sep 17 00:00:00 2001 From: LaraStuStu Date: Sat, 28 Mar 2020 14:01:36 +0800 Subject: [PATCH] Create util.py --- .../tests/labelme_tests/utils_tests/util.py | 37 +++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 DataAnnotation/labelme/tests/labelme_tests/utils_tests/util.py diff --git a/DataAnnotation/labelme/tests/labelme_tests/utils_tests/util.py b/DataAnnotation/labelme/tests/labelme_tests/utils_tests/util.py new file mode 100644 index 0000000..5045811 --- /dev/null +++ b/DataAnnotation/labelme/tests/labelme_tests/utils_tests/util.py @@ -0,0 +1,37 @@ +import json +import os.path as osp + +from labelme.utils import image as image_module +from labelme.utils import shape as shape_module + + +here = osp.dirname(osp.abspath(__file__)) +data_dir = osp.join(here, '../data') + + +def get_img_and_data(): + json_file = osp.join(data_dir, 'apc2016_obj3.json') + data = json.load(open(json_file)) + img_b64 = data['imageData'] + img = image_module.img_b64_to_arr(img_b64) + return img, data + + +def get_img_and_lbl(): + img, data = get_img_and_data() + + label_name_to_value = {'__background__': 0} + for shape in data['shapes']: + label_name = shape['label'] + label_value = len(label_name_to_value) + label_name_to_value[label_name] = label_value + + n_labels = max(label_name_to_value.values()) + 1 + label_names = [None] * n_labels + for label_name, label_value in label_name_to_value.items(): + label_names[label_value] = label_name + + lbl = shape_module.shapes_to_label( + img.shape, data['shapes'], label_name_to_value + ) + return img, lbl, label_names -- GitLab