python實現對文件中圖片生成帶標簽的txt文件方法
更新時間:2018年04月27日 10:35:51 作者:Xiaoli_Coding
下面小編就為大家分享一篇python實現對文件中圖片生成帶標簽的txt文件方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
在深度學習中經常需要生成帶標簽的圖片名稱列表,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實現對文件中圖片生成帶標簽的txt文件方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
獲取CSDN文章內容并轉換為markdown文本的python
這篇文章主要介紹了自己寫的小工具,可以直接獲取csdn文章并轉換為markdown格式,需要的朋友可以參考下2020-09-09

