-- coding: utf-8 --
import os
def file_name(file_dir):
for root, dirs, files in os.walk(file_dir):
print(root) #当前目录路径
print(dirs) #当前路径下所有子目录
print(files) #当前路径下所有非目录子文件
import os
para=r"D:\Caffe-fangxu\caffe-master\examples\twolable\train\n02085936-Maltese_dog"
lable="3"
def file_name(file_dir):
for root,dirs,files in os.walk(file_dir):
#print(root) #当前目录路径
# print(dirs) #当前路径下所有子目录
return [files,root]#当前路径下所有非目录子文件
if __name__ == '__main__':
file=file_name(para)
f=open(file[1]+"\\test.txt","w")
for x in file[0]:
f.write(x+" "+lable+"\n")
f.close()
网友评论