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