ubuntu16.04+faster_rcnn+voc2007數據集訓練

前言

之前寫過一片博客是在Windows下配置faster_rcnn demo環境 ,如果在Windows上訓練數據集會出現很多問題,於是纔不得不轉入Ubuntu系統,現在發覺Ubuntu系統越來越好使。

1.配置caffe環境

  1. 首先在Ubuntu系統下配置caffe,可以參考這篇博客ubuntu16.04+anaconda2+gpu+caffe安裝
  2. 如果按照上述博客配置好了caffe的話,先不要高興太早,因爲你電腦上的cudnn版本很可能與faster_rcnn要求不一致,所以得重新配置caffe,但是上一步的步驟還是要進行的。
  3. 下載faster_rcnn代碼:鏈接: https://pan.baidu.com/s/1VQ7pGNB_pz9v12APdQ0QKA 提取碼: fi7p
    下載後解壓得到caffe-fast-rcnn文件夾
    在這裏插入圖片描述

參考這篇博客解決cudnn版本不一致問題中的第3步驟,就是將所有cudnn版本替換,然後重新配置caffe就行。最後不要忘了在變量環境中將原先版本的caffe替換爲caffe-fast-rcnn版本路徑

2.配置demo

  1. 如果你的caffe編譯好的話,這一步就很簡單了,先下載模型。

    cd py-faster-rcnn/data/scripts
    

    找到fetch_faster_rcnn_models.sh文件打開裏面有模型下載鏈接,不要使用Ubuntu系統命令下載,因爲根本下載不下來,建議在Windows上使用迅雷下載,但是好像需要會員才能下載下來,於是我這提供百度雲鏈接:鏈接: https://pan.baidu.com/s/1ItMcQq-3B2RuPhQaXRvYsw 提取碼: irya
    下載下來後複製到data文件夾下即可。
    在這裏插入圖片描述2. 進入tools文件夾,打開終端,輸入命令

    python demo.py
    

    如果出現結果則表示demo配置成功

3.訓練數據集

  1. 下載voc數據集:
    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
    下載下來後解壓成一個文件夾重命名爲VOCdevkit2007,放在data文件夾下,如下圖所示
    在這裏插入圖片描述如果你要是訓練自己的數據集,只需按照voc數據集格式製作就行,在這裏就不在敘述了。
  2. 下載imagenet_models模型,在py-faster-rcnn/data/scripts文件夾下找到fetch_imagenet_models.sh文件,同理在Windows下下載,這裏依然提供百度雲鏈接: https://pan.baidu.com/s/1yCMiOLZ0AHBRThPB-UUzJg 提取碼: g2mw
    下載下來後解壓到data文件夾下即可
    在這裏插入圖片描述

3.如果你要是訓練自己的數據集參考這篇博客修改網絡對應的參數訓練自己的數據
4. 開始訓練
不要參考第3步博客的訓練命令會報錯

 cd py-faster-rcnn
./experiments/scripts/faster_rcnn_end2end.sh 0 VGG_CNN_M_1024 pascal_voc

換一種訓練命令

chmod u+x /home/csm/py-faster-rcnn/tools/train_faster_rcnn_alt_opt.py

一定要加入chmod u+x命令否則會報錯沒有訪問權限,然後輸入命令

./tools/train_faster_rcnn_alt_opt.py --gpu 0 --net_name ZF --weights data/imagenet_models/ZF.v2.caffemodel --imdb voc_2007_trainval --cfg experiments/cfgs/faster_rcnn_alt_opt.yml

4.測試數據

如果上述訓練數據正確的話,會在output文件夾中保存訓練模型
輸入命令

chmod u+x tools/test_net.py
./tools/test_net.py --gpu 0 --def models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt --net output/faster_rcnn_alt_opt/voc_2007_trainval/ZF_faster_rcnn_final.caffemodel--imdb voc_2007_test --cfg experiments/cfgs/faster_rcnn_alt_opt.yml
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章