在深度學習中經常需要生成帶標簽的圖片名稱列表，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.

以上這篇python實現對文件中圖片生成帶標簽的txt文件方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。