python實現對檔案中圖片產生帶標籤的txt檔案方法

下面為大家分享一篇python實現對檔案中圖片產生帶標籤的txt檔案方法，具有很好的參考價值，希望對大家有所協助。一起過來看看吧

在深度學習中經常需要產生帶標籤的圖片名稱列表，xxxlist.txt檔案，下面寫一個簡單的python指令碼產生該檔案清單。

import osdef generate(dir,label):  files = os.listdir(dir)  files.sort()  print '****************'  print 'input :',dir  print 'start...'  listText = open(dir+'\\'+'list.txt','w')  for file in files:    fileType = os.path.split(file)    if fileType[1] == '.txt':      continue        name = file + ' ' + str(int(label)) +'\n'    listText.write(name)  listText.close()  print 'down!'  print '****************'  if __name__ == '__main__':   generate('C:\\Users\\Desktop\\image',1)

運行該指令碼，會在image檔案夾中產生一個list.txt檔案，並且每張圖片帶有標籤1.