Python创建cvs文件，包含标签和图片数据

　　  在深度学习或者机器学习的时候，常常需要对数据进行整理和分类，最常见的是通过对数据路径和标签写入

到一个整合的txt或者csv文件中，训练进行读取。

   

#coding=utf-8
#!/usr/bin/env python
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:
#
#  .
#  |-- README
#  |-- s1
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  |-- s2
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  ...
#  |-- s40
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#

if __name__ == "__main__":

    if len(sys.argv) != 3:
        print "usage: create_csv <base_path>"
        sys.exit(1)

    BASE_PATH=sys.argv[1]
    SEPARATOR=";"
    fh = open(sys.argv[2], '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()

　　调用方式如下

　　

python create.py d:/test/jaf d:/a.csv

　　生成文件结果如下图所示

　　通过替换可以将斜杠调整一致