caffe製作LMDB資料來源(十四)
阿新 • • 發佈:2019-02-20
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資料來源了。