1. 程式人生 > >caffe訓練自己的模型步驟

caffe訓練自己的模型步驟

1.準備資料

建立目錄mkdir mydata,mkdir mydata/train,mkdir mydata/val,在train下放訓練資料,在val下放驗證資料。

生成train.txt、val.txt、test.txt

1.1生成val.txt:find  -name *.png |grep -v train | cut -d/ -f3>val.txt

1.2生成train.txt:find  -name *.png |grep  train | cut -d/ -f3-4 > train.txt

新增分類:注意分類和值之間只能新增一個空格,否則執行create_imagenet.sh會報類似如下錯誤:

E0818 16:36:51.530107  9428 io.cpp:80] Could not open or find file /home/caffe/caffe-master/.../train_zc_1072.png 

1.3新增分類

sed -i '1,1200s/.*/& 0/' train.txt和sed -i'1201,2400s/.*/& 1/' train.txt。

val類似,但是val以為是兩類,所以我一開始放到了2個檔案,採用生成val.txt的方法,生成兩個檔案以後再合併,再新增分類。

也可以新增分類後再合併,合併如下:

vi 檔案1,檔案2,複製命令:1,&y,然後:bp來轉換到下一個檔案,然後P,就會貼上到游標處了。