python实现生成图片路径和对应标签

心得：网上看了太多的教程，都是相互抄袭。不自己动手，怎么能行?

文件夹列表：

生成的结果展示：

代码：

# coding=utf-8
# !/usr/bin/env python
import sys
import os.path

if __name__ == "__main__":

     #只需要将BASE_PATH修改为你的路径
     BASE_PATH='C:\\Users\\zhang\Desktop\\lfw_test\\lfw_test'
     SEPARATOR=" "
     fh = open('1.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()