Spaces:
Configuration error
Configuration error
| """ | |
| Copyright (c) Microsoft Corporation. | |
| Licensed under the MIT license. | |
| """ | |
| import os | |
| import os.path as op | |
| import numpy as np | |
| import base64 | |
| import cv2 | |
| import yaml | |
| from collections import OrderedDict | |
| def img_from_base64(imagestring): | |
| try: | |
| jpgbytestring = base64.b64decode(imagestring) | |
| nparr = np.frombuffer(jpgbytestring, np.uint8) | |
| r = cv2.imdecode(nparr, cv2.IMREAD_COLOR) | |
| return r | |
| except: | |
| return None | |
| def load_labelmap(labelmap_file): | |
| label_dict = None | |
| if labelmap_file is not None and op.isfile(labelmap_file): | |
| label_dict = OrderedDict() | |
| with open(labelmap_file, 'r') as fp: | |
| for line in fp: | |
| label = line.strip().split('\t')[0] | |
| if label in label_dict: | |
| raise ValueError("Duplicate label " + label + " in labelmap.") | |
| else: | |
| label_dict[label] = len(label_dict) | |
| return label_dict | |
| def load_shuffle_file(shuf_file): | |
| shuf_list = None | |
| if shuf_file is not None: | |
| with open(shuf_file, 'r') as fp: | |
| shuf_list = [] | |
| for i in fp: | |
| shuf_list.append(int(i.strip())) | |
| return shuf_list | |
| def load_box_shuffle_file(shuf_file): | |
| if shuf_file is not None: | |
| with open(shuf_file, 'r') as fp: | |
| img_shuf_list = [] | |
| box_shuf_list = [] | |
| for i in fp: | |
| idx = [int(_) for _ in i.strip().split('\t')] | |
| img_shuf_list.append(idx[0]) | |
| box_shuf_list.append(idx[1]) | |
| return [img_shuf_list, box_shuf_list] | |
| return None | |
| def load_from_yaml_file(file_name): | |
| with open(file_name, 'r') as fp: | |
| return yaml.load(fp, Loader=yaml.CLoader) | |