在深度學習中經常需要生成帶標簽的圖片名稱列表，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文件方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持VEVB武林網。
新聞熱點
疑難解答
