python實現(xiàn)對文件中圖片生成帶標(biāo)簽的txt文件方法
更新時間:2018年04月27日 10:35:51 作者:Xiaoli_Coding
下面小編就為大家分享一篇python實現(xiàn)對文件中圖片生成帶標(biāo)簽的txt文件方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
在深度學(xué)習(xí)中經(jīng)常需要生成帶標(biāo)簽的圖片名稱列表,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)
運(yùn)行該腳本,會在image文件夾中生成一個list.txt文件,并且每張圖片帶有標(biāo)簽1.
以上這篇python實現(xiàn)對文件中圖片生成帶標(biāo)簽的txt文件方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
獲取CSDN文章內(nèi)容并轉(zhuǎn)換為markdown文本的python
這篇文章主要介紹了自己寫的小工具,可以直接獲取csdn文章并轉(zhuǎn)換為markdown格式,需要的朋友可以參考下2020-09-09