Python 얼굴 인식 - csv 파일 만들기createcsv.py 코드 Python3.7

이것은 매우 작은 스크립트입니다. 유사한 차원 구조를 가진 얼굴 데이터베이스에서 CSV 파일을 만드는 데 도움을 줄 수 있습니다

import sys
import os.path

# This is a tiny script to help you creating a CSV file from a face
# database with a similar hierarchie:
#
#  philipp@mango:~/facerec/data/at$ tree
#  .
#  |-- README
#  |-- s1
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  |-- s2
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  ...
#  |-- s40
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#

if __name__ == "__main__":

    #if len(sys.argv) != 2:
    #    print "usage: create_csv "
    #    sys.exit(1)

    #BASE_PATH=sys.argv[1]
    BASE_PATH="D:\Downloads\orl databases"
    
    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("
")      
            label = label + 1
    fh.close()