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("\n")
             label = label + 1
     fh.close()

　　调用方式如下

　　

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

　　生成文件结果如下图所示

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