1. 程式人生 > >caffe製作LMDB資料來源(十四)

caffe製作LMDB資料來源(十四)

caffe製作LMDB資料來源

caffe的執行依賴於一個指令碼檔案,這個指令碼檔案就是LMDB資料來源的配置檔案。

配置檔案如下:


其實大多數地方是一樣的,只是要改少數部分。

EXAMPLE=/home/green/caffe-training
DATA=/home/green/caffe-training
TOOLS=//home/green/caffe/build/tools


TRAIN_DATA_ROOT=/home/tyd/下載/face_detect/train/


VAL_DATA_ROOT=/home/tyd/下載/face_detect/val/

要改這兩個地方。

example和data就是這個這個檔案存放的地方,如果訓練資料和測試資料都放在一起,那就可以直接指定這個目錄了。

下面的的TOOLS後面的三個目錄其實是固定的。

TRAIN_DATA_ROOT就是訓練資料所在的地方。

VAL_DATA_ROOT就是測試資料所在的地方。

train資料夾下就是這樣的:

如果識別是兩個分類,就0,1。多個分類就0,1,2,3,4等等下去。

最後還要改一下

可以看到有兩個檔案:DATA/train.txt和DATA/val.txt這個兩個檔案就是訓練的檔案了,要使用這兩個檔案來製作LMDB資料來源。

這是train.txt檔案的格式,前面是檔案的位置,後面是他的label。剛剛分了0/1這兩個檔案,現在就要拿到路徑了。

之後就是執行sh 剛剛檔案的名字.sh來製作LMDB資料來源了。