如何写数据集标签txt文件

情况：网上下载了一个人脸反欺诈（CASIA-CeFA），但是里面的标签文件只有单模态的。所以根据文件存储的目录写了一个包含多模态的数据集标签文件

文件目录路径

CASIA-CeFA
    |
    |——phase1
    |    |
    |    |——train—— 1_000_1_1_1  —— profile
    |    |              |...     |——depth
    |    |——dev         |...     |——ir
    |                           
    |
    |——phase2
         |
         |——test

上代码

import os
# train文件

# 绝对路径
absolute_path = r'E:\Dataset'
# 相对路径
relative_path = r'CASIA-CeFA\phase1\train'

load_path = os.path.join(absolute_path,relative_path)

# 存储文件
save_path = r'E:\Dataset\CASIA-CeFA\phase1\train_list.txt'

for dir_num in os.listdir(load_path):
    dir_path1 = os.path.join(load_path, dir_num)
    dir_path2 = os.path.join(relative_path, dir_num)
    a = []  # 存放多模态数据
    for modal_name in os.listdir(dir_path1):
        a.append([])
        modal_path1 = os.path.join(dir_path1, modal_name)
        modal_path2 = os.path.join(dir_path2, modal_name)
        for file_name in os.listdir(modal_path1):
            file_path = os.path.join(modal_path2, file_name)
            a[-1].append(file_path)
    # 存储文件路径及数据标签        
    with open(save_path, 'a') as f:
        for i in range(len(a[0])):
            f.write(a[2][i] + ' ')      # profile
            f.write(a[0][i] + ' ')      # depth
            f.write(a[1][i])            # ir

            # 判断标签条件
            if('3_1_4' in a[0][i]):
                f.write(' 0')
            else:
                f.write(' 1')
            f.write('\n')

结果图