1. 程式人生 > >快速配置caffe-ssd及如何配置環境變數路徑

快速配置caffe-ssd及如何配置環境變數路徑

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