caffe訓練自己的模型步驟
阿新 • • 發佈:2019-01-26
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,就會貼上到游標處了。