[opencv][python][scikit-learn]att_faces数据人脸识别

数据整理:

用Python的os模块进行遍历文件夹,用string.encode("utf-8")解决编码不能写入的问题用Python的csv模块写入csv文件    

#!/usr/bin python2.7# coding=utf-8import os.pathimport csvwith open('/home/lvsolo/下载/att_faces/list.csv', 'a+') as csvfile:    fieldnames = ['path', 'label']    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)    writer.writeheader()BASE_PATH = u"/home/lvsolo/下载/att_faces"SEPARATOR = ";"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)).encode('utf-8')            if subject_path[-2] == 's':                label = int(subject_path[-1])            else:                ss=subject_path[-2:]                label = int(subject_path[-2:])            with open('/home/lvsolo/下载/att_faces/list.csv', 'a+') as csvfile:                fieldnames = ['path', 'label']                writer = csv.DictWriter(csvfile, fieldnames=fieldnames)                writer.writerow({'path': abs_path, 'label': label})