Mac下跑仅CPU模式下的PVANET

Handsome胜

于 2017-08-01 10:29:57 发布

阅读量1.1k

点赞数 1

分类专栏： mac

本文链接：https://blog.csdn.net/Handsome_Sheng/article/details/76505859

版权

本文详细介绍了在Mac环境下，如何在CPU模式下搭建并运行PVANET。主要内容包括：安装Caffe（CPU版）、解决编译中遇到的cblas.h缺失、ProposalLayer重定义、nms.cpp错误及numpy头文件找不到等问题，以及调整配置文件以适应CPU运行。最后成功运行Demo并测试VOC dataset，得到与官方相同的精度。

摘要由CSDN通过智能技术生成

1、RCNN利用自带的caffe框架，但是编译所依赖的库和之前我的另一篇博文相同

详见我的博客Mac下安装Caffe—CPU ONLY

2、确认安装Cython easydict protobuf

3、pull pva-faster-rcnn

git clone –recursive https://github.com/sanghoon/pva-faster-rcnn.git

4、编译Cython modules

首先，打开.／lib/setup.py 注释掉和GPU相关

...
#CUDA = locate_cuda()
...
...
#self.set_executable('compiler_so', CUDA['nvcc'])
...
...
#Extension('nms.gpu_nms',
#[‘nms/nms_kernel.cu', 'nms/gpu_nms.pyx'],
#library_dirs=[CUDA['lib64']],
#libraries=['cudart'],
#language='c++',
#runtime_library_dirs=[CUDA['lib64']],
## this syntax is specific to this build system
## we're only going to use certain compiler args with nvcc and not with
## gcc the implementation of this trick is in customize_compiler() below
#extra_compile_args={'gcc': ["-Wno-unused-function"],
#’nvcc': ['-arch=sm_35',
#’—ptxas-options=-v',