win10编译Faster RCNN的cython_utils
安装cuda、cudnn
略
请确保设置CUDAHOME环境变量
安装MinGW
略
修改cython_utils/setup.py
将以下代码块中的
nvcc改为nvcc.exe
lib64改为lib
# first check if the CUDAHOME env variable is in use
if 'CUDAHOME' in os.environ:
home = os.environ['CUDAHOME']
nvcc = pjoin(home, 'bin', 'nvcc')
else:
# otherwise, search the PATH for NVCC
default_path = pjoin(os.sep, 'usr', 'local', 'cuda', 'bin')
nvcc = find_in_path('nvcc', os.environ['PATH'] + os.pathsep + default_path)
if nvcc is None:
raise EnvironmentError('The nvcc binary could not be '
'located in your $PATH. Either add it to your path, or set $CUDAHOME')
home = os.path.dirname(os.path.dirname(nvcc))
cudaconfig = {'home':home, 'nvcc':nvcc,
'include': pjoin(home, 'include'),
'lib64': pjoin(home, 'lib64')}
修改cython_utils/Makefile
修改 rm -rf 为 rd/del
all:
python setup.py build_ext --inplace
rm -rf build
clean:
rm -rf */*.pyc
rm -rf */*.so
## 新的改变
编译
于cython_utils下,执行 mingw32-make
生成bbox和nms计算脚本如下: