ubuntu环境 编译CPU版的NMS

关键点检测
 

第一步：下载源码
git clone –recursive https://github.com/zigangzhao-ai/Pose-key-points-detection.git

第二步：编译cpython模块
进入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’,**
# ‘-c’,
# ‘–compiler-options’,
# “’-fPIC’”]},
# include_dirs = [numpy_include, CUDA[‘include’]]
# ),
…
修改完之后保存退出，在lib目录下执行 make命令。

第三步：修改nms.py编译文件
注释掉：#from .gpu_nms import gpu_nms

运行python setup.py build_ext --inplace

生成.so文件即可。