vim create_db.py

# -*- 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")
 #查看圖片目錄下的文件,相當于shell指令ls
 images_name = os.listdir(images_path)
 #遍歷所有文件名
 for eachname in images_name:
 #正則表達式這里可以根據(jù)情況進行更改，如果多類記得仿照下面定義相關變量
 #正則表達式規(guī)則:找以cat開頭,緊跟0到49個數(shù)字,并以jpg結(jié)尾的圖片文件
 pattern_cat = r'(^cat\d{0,49}.jpg$)'
 #正則表達式規(guī)則:找以bird開頭,緊跟0到49個數(shù)字,以jpg結(jié)尾的圖片文件
 pattern_bird = r'(^bird\d{0,49}.jpg$)'#更改變量名需要定義
 #正則表達式匹配
 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便是。當然相關變量也要記得定義。
 #打印成功信息
 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)