Spaces:
Configuration error
Configuration error
File size: 1,747 Bytes
0034848 |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
"""
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)
|