import os
gPath = os.walk("/home/") #图片的路径
list_file = open('/home/train40.txt', 'w') #存取路径的txt文件的路径
for path,d,filelist in gPath:
for filename in filelist:
if filename.endswith('jpg'): #图片的格式
list_file.write((path+'/'+filename)+'\n')
list_file.close()