學了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,即可。