py-faster-rcnn安装及训练自己的数据集

15 篇文章 4 订阅
4 篇文章 0 订阅

一、硬件和环境:

  1. GTX 1080 Ti, Cuda 9.0, Cudnn 7.6.0
  2. Ubuntu16.04
  3. Anaconda3 caffe3虚拟环境
    (caffe3虚拟环境用pip install opencv-python和pip install opencv-contrib-python,然后下载caffe并成功编译。)
  4. https://blog.csdn.net/la_fe_/article/details/88547816
  5. https://github.com/rbgirshick/py-faster-rcnn

二、下载 py-faster-rcnn并修改Makefile.config文件

source activate caffe3
cd  workspace/faster-rcnn/
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
cd py-faster-rcnn/
cd lib/
make
cd ..
cd caffe-fast-rcnn/
vi Makefile.config

将在workspace/caffe/Makefile.config和Makefile文件复制到workspace/faster-rcnn/py-faster-rcnn/caffe-fast-rcnn/文件夹下,其中Makefile不改变,而Makefile.config文件根据视频里面的Makefile.config文件做一些修改,如

  1. 注释掉USE_CUDNN:=1
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 

改为:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

在这里插入图片描述在这里插入图片描述在这里插入图片描述在这里插入图片描述在这里插入图片描述在这里插入图片描述

三、修改Makefile

vi Makefile

在这里插入图片描述

四、编译py-faster-rcnn

make -j8
make pycaffe

打开workspace/faster-rcnn/py-faster-rcnn/data/scripts/fetch_faster_rcnn_models.sh,复制里面的网址https://dl.dropboxusercontent.com/s/o6ii098bu51d139/faster_rcnn_models.tgz?dl=0下载faster-rcnn_models
在这里插入图片描述下载好之后把模型放在py-faster-rcnn/data/文件夹下。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值