faster rcnn训练(一)跑通原生模型

基于ubuntu16.04  caffe  python

一、下载文件

在Ubuntu根目下运行命令行的下载:

  git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

二、编译caffe

1.进入py-faster-rcnn/lib   make

2.进入py-faster-rcnn/caffe-fast-rcnn    复制 Makefile.config.example       Rename为       Makefile.config

修改参数

 
WITH_PYTHON_LAYER := 1
USE_CUDNN := 1

3.由于caffe版本不同的问题,将caffe下的文件拷贝或替换入caffe-faster-rcnn:

include/caffe/util/cudnn.hpp 

include/caffe/layers  所有以cudnn开头的文件

src/caffe/layer  所有以cudnn开头的文件

4.编译caffe

进入py-faster-rcnn/caffe-faster-rcnn

  make -j64


编译成功后

make pycaffe

三、下载训练好的模型

1.进入py-faster-rcnn  下载faster-rcnn models trained

./data/scripts/fetch_fater_rcnn_models.sh

2.下载caffe models(ZF、VGG16等)

./data/scripts/fetch_imagenet_modelsl.sh

四、跑通demo.py

进入py-faster-rcnn

./tools/demo.py  - - gpu 0





  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值