当前位置:   article > 正文

Mxnet---faster-rcnn环境安装_mxnetbinary faster rcnn

mxnetbinary faster rcnn

官方版本已经有faster-rcnn的实现,所以这里从官方下载mxnet并安装编译。以下版本的安装是对于mxnet 0.94版本,0.95版本的直接编译就可以用faster-rcnn。

安装环境:

  • Ubuntu14.04
  • GPU gtx1080

1.mxnet的安装
这里还是从两种情况分析,一种是之前已经安装好了mxnet,但是没有安装编译faster-rcnn,第二种是没有安装过任何mxnet。
对于第一种情况需要做的是,首先是要知道有关rcnn的文件在文件夹mxnet/example/rcnn中,你需要做的步骤是:
- 复制rcnn中的operator文件中的文件到mxnet/src/operator/
- 重新编译mxnet,在mxnet根目录中:make -j$(nproc)
- 编译python接口: cd python然后sudo python setup.py install

以上完成以后,命令窗口输入:

$ python
>>> import mxnet as mx
>>>
  • 1
  • 2
  • 3

出现以上说明成功。
第二种情况其实和第一种类似,前提还是要下载mxnet,只是到了编译部分需要把mxnet/example/rann/operator中的文件先复制到mxnet/src/operator,剩下的按照官方教程做即可

2.下载预训练模型和数据集

example/rcnn$: bash script/additional_deps.sh
#Download training data
example/rcnn$: bash script/get_voc.sh
#Download pretrained model
example/rcnn$: bash script/get_pretrained_model.sh
#Training and testing
example/rcnn$: bash script/vgg_voc07.sh 0,1 (use gpu 0 and 1)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

执行完之后会在rcnn文件夹中多出data和model两个文件夹。

3.测试
命令:mxnet/example/rcnn$ python demo.py --prefix final --epoch 0 --image myimage.jpg --gpu 0

说明: final 是最后训练好的模型,在model文件中是e2e-0010.params文件,可以重命名也可以修改final为e2e-0010, 但是final的完整名字是final-0000.params,所以为e2e-0010-0000.params,最后放到rnn目录中, myinage.jpg为需要检测的图片。如果要显示图片可以加参数“- -vis”

这里写图片描述

参考文献
1.http://mxnet.io/tutorials/computer_vision/detection.html
2.https://github.com/dmlc/mxnet/tree/master/example/rcnn

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/250180
推荐阅读
相关标签
  

闽ICP备14008679号