1.數據準備,我們需要一個train和valid文件夾,一個train.txt和val.txt(圖片文件夾的位置可以任意,但是這兩個txt文件的位置在data/fingervein/目錄下)
train和valid文件夾自然就存放著待訓練的圖片,train.txt和val.txt的數據格式如下
?
FV_USM_beforeLMDB/train_example/1/1.jpg 0
FV_USM_beforeLMDB/train_example/1/2.jpg 0
FV_USM_beforeLMDB/train_example/1/3.jpg 0
FV_USM_beforeLMDB/train_example/1/4.jpg 0
FV_USM_beforeLMDB/train_example/1/5.jpg 0
FV_USM_beforeLMDB/train_example/1/6.jpg 0
FV_USM_beforeLMDB/train_example/2/1.jpg 1
FV_USM_beforeLMDB/train_example/2/2.jpg 1
FV_USM_beforeLMDB/train_example/2/3.jpg 1
FV_USM_beforeLMDB/train_example/2/4.jpg 1
FV_USM_beforeLMDB/train_example/2/5.jpg 1
FV_USM_beforeLMDB/train_example/2/6.jpg 1
FV_USM_beforeLMDB/train_example/3/1.jpg 2
FV_USM_beforeLMDB/train_example/3/2.jpg 2
FV_USM_beforeLMDB/train_example/3/3.jpg 2
前邊是圖片路徑,后便是label,注意label是從0開始的。
?
2. 在example目錄下創建自己的文件夾fingervein,將imagenet中所有的文件烤過來,然后進行修改。
create_imagenet.sh
EXAMPLE=/home/wuchao/caffe/examples/fingervein
//存放train.txt和val.txt的目錄
DATA=/home/wuchao/caffe/data/fingervein
//不改動
TOOLS=/home/wuchao/caffe/build/tools
//圖片存放的目錄
TRAIN_DATA_ROOT=/home/wuchao/caffe/MyFiles/
VAL_DATA_ROOT=/home/wuchao/caffe/MyFiles/
運行create_imagenet.sh,生成兩個數據文件lmdb