一、安裝caffe
官網安裝步驟:http://caffe.berkeleyvision.org/install_apt.html
1.依賴環境安裝:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler sudo apt-get install --no-install-recommends libboost-all-d
2.安裝blas
sudo apt-get install libopenblas-dev
3.python安裝
sudo apt-get install python-dev
4.其他庫安裝
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
二、安裝caffe-ssd
git clone https://github.com/weiliu89/caffe.git cd caffe git checkout ssd
編譯:
# Modify Makefile.config according to your Caffe installation. cp Makefile.config.example Makefile.config make -j8 # Make sure to include $CAFFE_ROOT/python to your PYTHONPATH. make py make test -j8 # (Optional) make runtest -j8
如果過程中報錯,嘗試修改一下文件:
Makefile.config:
將USE_CUDNN :=1取消註釋
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include後面打上一個空格, 然後添加/usr/include/hdf5/serial, 如果沒有這一句會報一個找不到hdf5.h的錯誤
替換NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS) 爲NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS) 保存退出。
模型下載:
遇到的問題:
1、提示沒有Caffe模塊:
執行的操作
echo "export PYTHONPATH=/home/xiaoda/caffe/python" >> ~/.bashrc source ~/.bashrc echo $PYTHONPATH #檢查環境變量的值
2、模塊沒有定義輸出函數
xiaoda@jdh-Precision-7920-Tower:~/caffe$ python examples/ssd/score_ssd_pascal.py Traceback (most recent call last): File "examples/ssd/score_ssd_pascal.py", line 2, in <module> import caffe File "/home/xiaoda/caffe/python/caffe/__init__.py", line 1, in <module> from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver File "/home/xiaoda/caffe/python/caffe/pycaffe.py", line 13, in <module> from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \ ImportError: dynamic module does not define module export function (PyInit__caffe)
可能是python3.6.5的問題,因爲版本較高的 Python 和 caffe 不一定兼容;caffe 官網 (http://caffe.berkeleyvision.org/ ) 上也提示說, 只是較好的支持 caffe 2.7 版本;
ubuntu16默認安裝python2 和python3,
#檢查python版本: python --version
#設置默認python: echo alias python= python2 >> ~/.bashrc
提示缺啥,安裝啥
sudo apt-get install python-skimage sudo apt-get install python-protobuf
大工告成: