在进行人脸识别模型训练的时候,需要把人脸图片的路径和图片的标签表示出来,这个时候就可以通过python来完成这项工作了
import sys
import os.path
BASE_PATH="D:/百度网盘下载/64_CASIA-FaceV5/CASIA-FaceV5 (000-099)"
# 图片所在的文件夹
SEPARATOR=";"
# 项目分隔符
fh = open("at.txt",'w')
# 生成文件的路径和名称
label = 0
for dirname, dirnames, filenames in os.walk(BASE_PATH):
for subdirname in dirnames:
subject_path = os.path.join(dirname, subdirname)
for filename in os.listdir(subject_path):
abs_path = "%s/%s" % (subject_path, filename)
print("%s%s%d" % (abs_path, SEPARATOR, label))
fh.write(abs_path)
fh.write(SEPARATOR)
fh.write(str(label))
fh.write("\n")
label = label + 1
fh.close()