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()