雾聪
2024-10-10 4d8e96f695445af61e905d67fa486fcb382ec4ac
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
import csv
import os
 
import logging
 
 
def get_abs_path(rel_path):
    """
    Get absolute path
 
    Args:
        rel_path: relative path to this file
 
    Returns absolute path
    """
    abs_path = os.path.dirname(os.path.abspath(__file__)) + os.sep + rel_path
 
    if not os.path.exists(abs_path):
        logging.warning(f"{abs_path} does not exist")
    return abs_path
 
 
def load_labels(abs_path):
    """
    loads relative path file as dictionary
 
    Args:
        abs_path: absolute path
 
    Returns dictionary of mappings
    """
    label_tsv = open(abs_path, encoding="utf-8")
    labels = list(csv.reader(label_tsv, delimiter="\t"))
    return labels