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()
作者:zhangyuexiang123