1、如果想要利用代碼（不論是python、c++、亦或是matlab）實現生成標簽文件，首先，也是灰常重要的一件事就是你的圖片集一定要是有規律的命名。數字字母順序排開。這一點非常重要，相關重命名方法請自行百度或請教大牛。

2、如圖為博主的圖片集。（注意命名規律）

博主只分為貓和鳥兩類。

3、看代碼。

在caffe根目錄下創建一個我們的工程目錄my-caffe-project

創建并編輯create_db.py文件，使用如下指令：

vim create_db.py

然后，代碼內容就是我們的重點了。注意相關變量和路徑的修改。

# -*- coding: UTF-8 -*-import osimport re"""函數說明:生成圖片列表清單txt文件Parameters: images_path - 圖片存放目錄 txt_save_path - 圖片列表清單txt文件的保存目錄Returns: 無Author: Jack CuiModify: 2017-03-29"""def createFileList(images_path, txt_save_path): #打開圖片列表清單txt文件 fw = open(txt_save_path,"w") #查看圖片目錄下的文件,相當于shell指令ls images_name = os.listdir(images_path) #遍歷所有文件名 for eachname in images_name: #正則表達式這里可以根據情況進行更改，如果多類記得仿照下面定義相關變量 #正則表達式規則:找以cat開頭,緊跟0到49個數字,并以jpg結尾的圖片文件 pattern_cat = r'(^cat/d{0,49}.jpg$)' #正則表達式規則:找以bird開頭,緊跟0到49個數字,以jpg結尾的圖片文件 pattern_bird = r'(^bird/d{0,49}.jpg$)'#更改變量名需要定義 #正則表達式匹配 cat_name = re.search(pattern_cat, eachname) bird_name = re.search(pattern_bird, eachname) #按照規則將內容寫入txt文件中 if cat_name != None:  fw.write(cat_name.group(0) + ' 1/n') if bird_name != None:  fw.write(bird_name.group(0) + ' 2/n')#如果分為多類，多加幾個if便是。當然相關變量也要記得定義。 #打印成功信息 print "生成txt文件成功" #關閉fw fw.close()#下面是相關變量定義的路徑if __name__ == '__main__': #caffe_root目錄 caffe_root = '/home/xn/caffe/' #my-caffe-project目錄 my_caffe_project = caffe_root + 'my-caffe-project/' #圖片存放目錄 images_path = caffe_root + 'data/mydata/myimagenet/train/' #生成的圖片列表清單txt文件名 txt_name = 'train.txt' #生成的圖片列表清單txt文件的保存目錄 txt_save_path = my_caffe_project + txt_name #生成txt文件 createFileList(images_path, txt_save_path)

完成。如果想要生成測試標簽文件類似。

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