基于docker的python faster-rcnn caffe环境搭建+提取目标特征实验

本文链接：https://blog.csdn.net/real_cactus/article/details/112850460

文章目录

起初是为了使用faster-rcnn的目标提取功能提取图片中的目标特征，供后续模型学习，有一个比较好的实现bottom up attention，其为py-faster-rcnn在 Visual Genome 数据集预训练的模型,官方py-faster-rcnn在COCO数据集预训练下仅识别80个class，bottom up attention可识别1600个object class以及400个属性类别。我也是基于这个项目改写来实现图像对象特征预提取的。

论文地址：https://panderson.me/up-down-attention/ （CVPR 2018）

1. 环境配置前言

caffe环境配置真的挺麻烦的。。关键是许多相关版本和当前服务器上的相差很大，去适配就更加麻烦了，我这里使用基于Docker的方法来配置，因为我一直是习惯在Docker内配置深度学习环境及开发的，使用docker开发非常清爽！！这边可以参考我以前写的两篇文章来配置docker环境：

同时这里有位前辈曾经也记录过caffe下bottom up attention (py-faster-rcnn)环境配置，这是直接在机器上安装的。

下面，先预览一下可以跑得通的环境版本配置（后面有时间再研究研究cuda9.0以上的版本是不是可以使用吧，最近在赶实验实在是没时间）

Ubuntu16.04系统
GTX xxx 显卡
cuda8.0+cuDNN5.1
无需安装opencv3！

2. 下载caffe镜像

就一行指令：docker pull bvlc/caffe:gpu

当然可以去https://hub.docker.com/search/?q=SSD%20caffe&type=image搜索想要的caffe版本以及各种docker镜像。

镜像拉下来后，我的启动指令是：

sudo docker run --runtime=nvidia -it --privileged=true -p 1xxxx:22 -v /home/xxx/tf-torch:/root/tf-torch --name env_caffe bvlc/caffe:gpu /bin/bash

这里解释一下，-p 1xxxx:22 是为了能通过1xxxx这个对外接口来直接ssh到运行了ssh服务的docker内，这对于远程开发调试来说十分有意义。-v是文件映射。

另外注意这个版本是基于python2.7的。

3. 下载bottom-up-attention代码以及编译

这是能成功执行的代码版本：https://github.com/realcactus/bottom-up-attention，用文章开头贴出的官方的版本应该也差不多，都是几年没更新代码了，不过我也没去细看，以防万一我还是放在这里。

然着项目README文件中的Installation部分来操作即可，lib中的make不会有任何问题，最后一步，Build Caffe and pycaffe这里还是会遇到问题的，也别看他写的注释了，这样来：

cd bottom-up-attention
cd caffe
// 将Makefile.config自己写一份出来
mv Makefile.config.example Makefile.config
// 这里需要注意的是，先只要把里面的
// WITH_PYTHON_LAYER := 1
// USE_NCCL := 1
// USE_CUDNN := 1
// 这三行前面的注释删掉就好，后面再有问题跟着来，直接放一个Makefile文件配置简直一脸懵逼


// 你需要安装python-opencv2，但是python2.7中安装需要注意，加上清华源，另外指定版本到3.2左右，新版本的，import时会报错。
pip install opencv-python==3.2.0.6  -i https://pypi.tuna.tsinghua.edu.cn/simple

// 接下来make
make -j8 && make pycaffe

// ok,会遇到问题，别慌，如果你用的是docker，我们应该是百分百一样的过程。
// 遇到问题，fatal error: hdf5.h: 没有那个文件或目录，这个是路径问题。参考解决：https://www.cnblogs.com/xiangfeidemengzhu/p/7058391.html
// 即locate hdf5.h，找到你的hdf5.h文件在哪，然后修改Makefile.config文件，下面那行中后半部分是我们要加上的
// INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/

// 修改Makefile文件
// LIBRARIES += glog gfla