caffe實現Alexnet網路結構(利用自己的資料)
阿新 • • 發佈:2019-01-22
1.新建測試集和校驗集的txt文件 文件內容為:路徑+標籤
2.修改網路引數,輸入資料格式改為圖片格式(預設格式為lmdb)
3.執行指令碼檔案,開始執行
備註:如果使用均值檔案,需要輸入圖片的大小為256*256;均值檔案我使用的是imagenet的均值檔案,並沒有自己生成。
1.新建測試集和校驗集的txt文件 文件內容為:路徑+標籤
1.1在data/目錄下,新建mydata資料夾;
1.2在mydata資料夾中新建兩個資料夾train var;
1.3在train資料夾中新建n個資料夾(n代表類別數,比如有1000類影象就新建1000個),每個資料夾中存放相同類別 的用於測試的影象庫檔案
1.4在var中存放用於校驗的影象庫檔案(不區分類別!!!)
1.5生成兩個txt文件(train.txt var.txt),內容為:路徑+標籤(建議採用全路徑;另外txt的生成我是用matlab實現 的,也可以利用caffe提供的指導文 檔)
2.修改網路引數,輸入資料格式改為圖片格式(預設格式為lmdb)
2.1在examples/目錄下新建mytask資料夾
2.2將caffe-master/model/bvlc_reference_caffenet中的所有檔案拷貝到mytask目錄下,並對train_val.prototxt進行修改;修 改如下
3.執行指令碼檔案,開始執行
sudo ./exmaples/mytask/train_caffenet.sh
因為我修改了solver.prototxt檔案,所以只迭代1000次就出結果啦~