def _get_slip_labels()

in pytouch/datasets/slip_sequence.py [0:0]


def _get_slip_labels(root, label_file_name="manual_frame.txt"):
    slip_label = []
    clip_list = None
    cnt = 0
    data_path = os.path.join(root, "data", label_file_name)

    for line in open(data_path, "r"):
        line = line.split(",")[0].split("    ")[-1]
        if cnt % 13 == 0:
            clip_list = []
        if (cnt % 13 > 0) and (cnt % 13 < 11):
            clip_list.append(int(line))
        if cnt % 13 == 11:
            slip_label.append(clip_list)
        cnt = cnt + 1
    slip_label = np.array(slip_label)
    return slip_label