快速配置caffe-ssd及如何配置環境變數路徑
阿新 • • 發佈:2018-12-12
caffe-ssd是目標檢測的一個經典模型,屬於單階段檢測模型,速度較快,本人實現過好幾次,也實現過ssd的改進版本fssd和DSSD,效果能提升幾個百分點。本篇部落格主要介紹caffe-ssd的配置流程以及如何設定環境變數路徑。
第一步:配置好caffe,包括NVIDIA-DIRVER,CUDA,CUDNN,OPENCV.可參考以下教程:
OK,以上工作做好下面開始配置caffe-ssd
第二步:
以上網址是ssd官網,可以直接下載也可以使用命令列:
git clone https://github.com/weiliu89/caffe/tree/ssd
接下來配置檔案,可參考以下網址改:
第三步:
make all -j8
make test
make runtest
涉及到許可權問題可在每行命令列前加 sudo
編譯成功再
sudo make pycaffe
成功就可以用了。
第四步:檢查caffe是否配置成功
python
import caffe
exit()
這樣表明caffe安裝成功了。
第五步,caffe-ssd資料準備,可參照以下網址:
第六步:設定環境變數路徑:
sudo gedit ~/.bashrc
export PYTHONPATH=/caffe路徑/python:$PYTHONPATH
source ~/.bashrc
在執行ssd_pascal.py時,出現如下錯誤:
網上查閱幾乎所有的人都會說我這是環境變數沒有設定好,可是我第六步明明做好了,百思不得其解,最後我開啟ssd_pascal.py加入了兩句話就ok了:
import sys
sys.path.append("caffe路徑/python)
參考以下連結:
至此,我的caffe-ssd就可以跑起來了!!
caffe-ssd預處理模型:
連結: https://pan.baidu.com/s/1frU5MPcNeg0GltgCrgTY9A 提取碼: ifwh