Caffe-SSD项目下载

git clone https://github.com/weiliu89/caffe.git caffe-ssd
cd caffe-ssd  &&  git checkout ssd

安装预备环境

preparation

需要提前配置好Opencv3

# 可通过该命令查看opencv版本
pkg-config --modversion opencv

setup

sudo apt-get update
sudo apt-get install software-properties-common
sudo add-apt-repository universe 
sudo add-apt-repository multiverse

Install dependencies

sudo apt-get install libboost-dev libboost-all-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev libatlas-base-dev liblmdb-dev libblas-dev libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler

Caffe-SSD配置文件修改

配置路径

# 在最后添加,caffe源码默认存放在当前用户的主目录下
echo 'export CAFFE_ROOT=/your/path/to/caffe-ssd' >> ~/.bashrc   # 配置$CAFFE_ROOT # 配置$CAFFE_ROOT
# 将/usr/lib/python2.7/dist-packages和$CAFFE_ROOT/python追加到$PYTHONPATH.
echo 'export PYTHONPATH=$PYTHONPATH:/usr/lib/python2.7/dist-packages:$CAFFE_ROOT/python'>>~/.bashrc
# 将$CAFFE_ROOT/build/tool命令工具追加到$PATH中
echo 'export PATH=$PATH:$CAFFE_ROOT/build/tool' >> ~/.bashrc
# 使环境变量生效
source ~/.bashrc

修改配置项

cd $CAFFE_ROOT
# 复制Makefile.config
cp Makefile.config.example Makefile.config
# 修改Makfile.config配置文件
sudo vim Makefile.config

具体修改内容包含：

第5行：
- # USE_CUDNN := 1
+USE_CUDNN := 1
第21行：
-# OPENCV_VERSION := 3
+OPENCV_VERSION := 3
第35行
-CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
-               -gencode arch=compute_20,code=sm_21 \
-               -gencode arch=compute_30,code=sm_30 \
-               -gencode arch=compute_35,code=sm_35 \
-               -gencode arch=compute_50,code=sm_50 \
-               -gencode arch=compute_52,code=sm_52 \
-               -gencode arch=compute_60,code=sm_60 \
-               -gencode arch=compute_61,code=sm_61 \
-               -gencode arch=compute_61,code=compute_61
+CUDA_ARCH := -gencode arch=compute_72,code=sm_72 \
+               -gencode arch=compute_72,code=compute_72
第85行
-# WITH_PYTHON_LAYER := 1
+ WITH_PYTHON_LAYER := 1

第92和93行
-INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
-LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
+INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
+LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/aarch64-linux-gnu /usr/lib/aarch64-linux-gnu/hdf5/serial

修改Makfile配置文件

sudo vim Makefile
-LIBRARIES += glog gflags protobuf boost_system boost_filesystem m
+LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial boost_regex

编译和测试

cd $CAFFE_ROOT
make -j32
make pycaffe # 编译pycaffe
make test -j8
# (可选)
make runtest -j8

注意

make runtest报错

参考 ：
https://devtalk.nvidia.com/default/topic/1044473/jetson-agx-xavier/caffe-installation-on-xavier/1
https://blog.csdn.net/weixin_38740463/article/details/81626402