採用caffe的model在自己的數據上面進行 fine-tuning

   學了caffe的基本構架以後,想採用caffe的model在自己的數據庫上面進行fine-tuning.

   我的主要工作還是做圖像分類,所以我沒有修改caffe的構架。

  (1)只是將最後的輸出1000改爲了14,同時把train和val中的fc8全部改爲一個自定義的名字,如fc-ft,這個名字隨

便自己定義。

  (2)用自帶的create_imagenet.sh腳本生成訓練和驗證數據,用make_imagenet_mean.sh生成均值文件;

  (3)編寫運行腳本,finetune.sh:並且注意自己的相關文件的路徑。   

#!/usr/bin/env sh

TOOLS=../../build/tools

GLOG_logtostderr=1 $TOOLS/finetune_net.bin std_solver.prototxt caffe_reference_imagenet_model 

echo "Done."
  (4)./finetune.sh,即可。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章