in src/utils.py [0:0]
def get_label_map(label_file):
label_map = {}
labels = open(label_file, 'r')
for line in labels:
line = line.rstrip("\n")
ids = line.split(',')
label_map[int(ids[0])] = ids[2]
return label_map