python實現(xiàn)對文件中圖片生成帶標簽的txt文件方法
更新時間:2018年04月27日 10:35:51 作者:Xiaoli_Coding
下面小編就為大家分享一篇python實現(xiàn)對文件中圖片生成帶標簽的txt文件方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
在深度學習中經(jīng)常需要生成帶標簽的圖片名稱列表,xxxlist.txt文件,下面寫一個簡單的python腳本生成該文件列表。
import os
def 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實現(xiàn)對文件中圖片生成帶標簽的txt文件方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
TensorFlow實現(xiàn)保存訓練模型為pd文件并恢復
今天小編就為大家分享一篇TensorFlow實現(xiàn)保存訓練模型為pd文件并恢復,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02

