情况:网上下载了一个人脸反欺诈(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')
结果图