1. 程式人生 > >fast-rcnn配置執行VGG16.caffemodel和VGG_CNN_M_1024.v2.caffemodel(Ubuntu14.04)

fast-rcnn配置執行VGG16.caffemodel和VGG_CNN_M_1024.v2.caffemodel(Ubuntu14.04)

在我的上個部落格中已經對fast-rcnn配置執行demo.py做出了相應說明,在本部落格中我將對fast-rcnn配置執行VGG16.caffemodel和VGG_CNN_M_1024.v2.caffemodel做出具體操作說明,希望可以解決大家在訓練vgg網路時出現的問題。

好了,我們來玩玩這個fast-rcnn配置執行VGG16.caffemodel和VGG_CNN_M_1024.v2.caffemodel

第一部分:下載資料test data and VOCdevkit

下載原網址:http://www.cs.berkeley.edu/~rbg/fast-rcnn-data(供選擇)

或者終端輸入:
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

第二部分:解壓生成資料夾 VOCdevkit
data
下有資料夾命名為VOCdevkit,此資料夾中包含VOC2007VOCcode(如第三部分所示)
就在data目錄下輸入下列命令解壓,解壓檔案會自動跑到devkit這個資料夾,其中VOC2007會自動包含trainvaltest這兩個檔案,即終端輸入:
tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar
tar xvf VOCdevkit_08-Jun-2007.tar

第三部分:資料夾包含目錄
$VOCdevkit/                          #development kit
$VOCdevkit/VOCcode/        #VOC utility code
$VOCdevkit/VOC2007         #image sets, annotations, etc.



第四部分:建立連結

終端輸入:
cd $FRCN_ROOT/data其中$表示您fast-rcnn的目錄位置
sudo ln -s VOCdevkit VOCdevkit2007

第五部分:下載 pre-computed Selective Search objectproposals
下載地址原網址:http://www.cs.berkeley.edu/~rbg/fast-rcnn-data(選擇selective search那個檔案)
拷貝放在data目錄下解壓第六部分:下載 pre-trained ImageNet models CaffeNet (model S), VGG_CNN_M_1024 (model M), and VGG16 (model L).

下載地址原網址:http://www.cs.berkeley.edu/~rbg/fast-rcnn-data(選擇imagenet_models)
拷貝放在data目錄下解壓第七部分:訓練測試如下:訓練Train a Fast R-CNN detector.
fast-rcnn目錄下終端執行:
./tools/train_net.py --gpu 0 --solver models/VGG16/solver.prototxt --weights data/imagenet_models/VGG16.v2.caffemodel
如果出現錯誤:
EnvironmentError: MATLAB command 'matlab' not found. Please add 'matlab' toyour PATH.
則增加(終端執行):

export PATH=$PATH:/usr/local/MATLAB/R2014a/bin(前提保證matlab裝好,make matcaffe通過)測試Test a Fast R-CNN detector.
./tools/test_net.py --gpu 0
(用0不能用1,如果用1會顯示核心已轉存,即是記憶體出錯) --def models/VGG16/test.prototxt --net output/default/voc_2007_trainval/vgg16_fast_rcnn_iter_40000.caffemodel

TIPS:
:如果用VGG_CNN_M_1024.v2.caffemodel這個模型,改一下中間的資料夾名稱,即將VGG16改為VGG_CNN_M_1024.v2.caffemodel即可。

任何問題請加唯一QQ2258205918(名稱samylee)!