运行caffe版(python)faster RCNN
2017-07-12 10:44
253 查看
第一步:安装python包依赖
第二步:下载源代码
sudo pip install Cython sudo pip install easydict
第二步:下载源代码
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git[/code]
在py-faster-rcnn/lib目录下修改和编译Cython
将setup.py文件第135行'nvcc': ['-arch=sm_35',
对照NVIDIA网站
改为符合自己显卡的数字,比如GTX 1080对应61,即'nvcc': ['-arch=sm_61',
然后编译:make -j8
第三步:在caffe-faster-rcnn目录下修改Makefile.config文件cp Makefile.config.example Makefile.config注意修改
USE_CUDNN := 1
WITH_PYTHON_LAYER := 1
USE_PKG_CONFIG := 1
第四步:替换为新版CUDNN文件
由于caffe-fast-rcnn在编写时使用的是低版本的cudnn,我们需要先下载或者git clone最新版的caffe,然后再进行文件替换。
主要替换文件如下:py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/cudnn.hpp py-faster-rcnn/caffe-fast-rcnn/src/caffe/util/cudnn.cpp py-faster-rcnn/caffe-fast-rcnn/src/caffe/layers/ 下16个cudnn_开头的文件 py-faster-rcnn/caffe-fast-rcnn/include/caffe/layers/下8个cudnn_开头的文件
第五步:编译make -j8 && make pycaffe
相关文章推荐
- 【Caffe】Python版本Faster R-CNN+ZF代码运行
- Ubuntu16.04+CUDA8.0+OpenCV3.1+python+caffe+faster-rcnn环境配置
- Window & ubuntu(非root用户)编译caffe并运行py-faster-rcnn
- Caffe: Faster-RCNN Python版本配置 (Windows)
- Caffe学习(八)运行py-faster-RCNN
- py-faster-rcnn(running the demo): ubuntu14.04+caffe+cuda7.5+cudnn5.1.3+python2.7环境搭建记录
- py-faster-rcnn之python引入_caffe.so
- Ubuntu16.04环境下安装caffe及运行faster-rcnn gpu 實現
- Caffe: Faster-RCNN Python版本配置 (Windows)
- py-faster-rcnn(caffe)配置和运行,训练自己的数据
- ubuntu14.04+CPU环境下搭建caffe,python接口以及运行fast-rcnn实例
- 深度学习之windows python faster rcnn 配置及demo运行
- 深度学习之windows python faster rcnn 配置及demo运行
- Ubuntu16.04环境下安装caffe及运行faster-rcnn
- Windows10环境下配置Caffe(Faster RCNN、python2.7)
- Caffe: Faster-RCNN Python版本配置 (Windows)
- win10 caffe python Faster-RCNN训练自己数据集(转)
- py-faster-rcnn运行实例部分问题
- 使用caffe训练faster-rcnn时遇到的问题总结
- 基于Faster-RCNN-TF的gpu运行总结(自己准备数据集)(训练篇)