1. 程式人生 > >caffe實現Alexnet網路結構(利用自己的資料)

caffe實現Alexnet網路結構(利用自己的資料)

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次就出結果啦~