Faster R-CNN-MXNet框架下的训练与测试

Faster R-CNN原文:https://arxiv.org/abs/1506.01497

代码:https://github.com/apache/incubator-mxnet/tree/master/example/rcnn

(Mxnet官方版本中对Faster R-CNN的实现)

Mxnet的安装:https://blog.csdn.net/zyxxlyj/article/details/80764460

1 下载预训练模型和数据集

进入example/rcnn文件夹

1) 安装附加依赖的环境,其中会因为超时多次中断

example/rcnn$:bash script/additional_deps.sh

2) 下载VOC数据集

example/rcnn$:bash script/get_voc.sh

包括:VOCtrainval_06-Nov-2007.tar

          VOCtest_06-Nov-2007.tar

          VOCtrainval_11-May-2012.tar

3) 下载预训练模型

example/rcnn$: bashscript/get_pretrained_model.sh

4) 训练及测试

example/rcnn$: bash script/vgg_voc07.sh 0,1(use gpu 0 and 1)

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

分别在数据集VOC07和VOC07+VOC12下训练(example/rcnn$: bash script/vgg_voc0712.sh 0,1),并在VOC07的测试集下测试,mAP分别为69.7%和76.1%(论文中为69.9%和73.2%)。

2 测试

mxnet/example/rcnn$python demo.py --prefix final --epoch 0 --image myimage.jpg --gpu 0

000704.jpg前者检测出person*2,horse*1,后者检测出person*2,horse*2,bicycle*1



评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值