Faster-rcnn模型下载
1、下载Faster-rcnn源码
caffe:
gitclone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
tensorflow:
git clone --recursive https://github.com/smallcorgi/Faster-RCNN_TF.git
2、编译
进入目录cd py-faster-rcnn/lib
make
出现错误:
python setup.py build_ext --inplace
Traceback (most recent call last):
File "setup.py", line 58, in <module>
CUDA = locate_cuda()
File "setup.py", line 46, in locate_cuda
raise EnvironmentError('The nvcc binary could not be '
EnvironmentError: The nvcc binary could not be located in your $PATH. Either add it to your path, or set $CUDAHOME
make: *** [all] Error 1
解决办法是打开setup.py文件将以下注释,即可:
#CUDA = locate_cuda()
#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']]
#),
#Extension(
# 'pycocotools._mask',
# sources=['pycocotools/maskApi.c', 'pycocotools/_mask.pyx'],
# include_dirs = [numpy_include, 'pycocotools'],
# extra_compile_args={
# 'gcc': ['-Wno-cpp', '-Wno-unused-function', '-std=c99']},
#),
再次make
3、下载预训练模型:
运行py-faster-rcnn/data/scripts下的fetch_faster_rcnn_models.sh脚本文件下载
但这里我没有下载成功,估计需要翻墙,在网上看见下载种子,亲测可用
针对caffe结构的:
http://academictorrents.com/details/cca56021739c8a75af3b58f536d4930266c25d5e/tech
针对tensorflow架构:
https://github.com/leihe001/tensorflow-vgg
https://github.com/leihe001/tensorflow-resnet
4、最后运行demo
./tools/demo.py
好像需要安装cuda才能运行,cpu下运行不通过,原因未知!
5、本人初学faster-rcnn,望各位指教,有意者可交流学习!
更多推荐
所有评论(0)