快捷導(dǎo)航

python方法生成txt標(biāo)簽文件的實例代碼

更新時間：2018年05月10日 08:43:19 作者：xunan003

今天小編就為大家分享一篇python方法生成txt標(biāo)簽文件的實例代碼，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

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

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

博主只分為貓和鳥兩類。

3、看代碼。

在caffe根目錄下創(chuàng)建一個我們的工程目錄my-caffe-project

創(chuàng)建并編輯create_db.py文件，使用如下指令：

vim create_db.py

然后，代碼內(nèi)容就是我們的重點(diǎn)了。注意相關(guān)變量和路徑的修改。

# -*- coding: UTF-8 -*-
import os
import re

"""
函數(shù)說明:生成圖片列表清單txt文件

Parameters:
 images_path - 圖片存放目錄
 txt_save_path - 圖片列表清單txt文件的保存目錄
Returns:
 無
Author:
 Jack Cui
Modify:
 2017-03-29
"""
def createFileList(images_path, txt_save_path):
 #打開圖片列表清單txt文件
 fw = open(txt_save_path,"w")
 #查看圖片目錄下的文件,相當(dāng)于shell指令ls
 images_name = os.listdir(images_path)
 #遍歷所有文件名
 for eachname in images_name:
 #正則表達(dá)式這里可以根據(jù)情況進(jìn)行更改，如果多類記得仿照下面定義相關(guān)變量
 #正則表達(dá)式規(guī)則:找以cat開頭,緊跟0到49個數(shù)字,并以jpg結(jié)尾的圖片文件
 pattern_cat = r'(^cat\d{0,49}.jpg$)'
 #正則表達(dá)式規(guī)則:找以bird開頭,緊跟0到49個數(shù)字,以jpg結(jié)尾的圖片文件
 pattern_bird = r'(^bird\d{0,49}.jpg$)'#更改變量名需要定義
 #正則表達(dá)式匹配
 cat_name = re.search(pattern_cat, eachname)
 bird_name = re.search(pattern_bird, eachname)
 #按照規(guī)則將內(nèi)容寫入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便是。當(dāng)然相關(guān)變量也要記得定義。
 #打印成功信息
 print "生成txt文件成功"
 #關(guān)閉fw
 fw.close()

#下面是相關(guān)變量定義的路徑
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)

完成。如果想要生成測試標(biāo)簽文件類似。

以上這篇python方法生成txt標(biāo)簽文件的實例代碼就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: