在Caffe中测试Faster R-CNN模型

简单介绍Caffe中如何测试Faster R-CNN训练出来的网络模型,得其mAP.

***********

一 准备测试集

1
data/VOCdevkit2007/VOC2007 文件夹中,将原有的文件重命名用以备份。将含有样本标记信息的.xml文件放入新建的Annotations文件夹,将含有测试样本名称(写成一列,名称后加一个空格,不带后缀,不留空行)信息的test.txt文件放入新建的ImageSets/Main文件夹,将测试样本图片放入新建的JPEGImages文件夹。

2
以上3个文件夹的样本信息必须对应,例如test.txt文件里的样本应能在其他两个文件夹中找到对应的.xml文件及图片。

***********

二 确定 py-faster-rcnn/tools/test_net.py 所需文件

1
确定模型测试对应网络的prototxt文件
例如:models/pascal_voc/VGG16/faster_rcnn_alt_opt/faster_rcnn_test.pt

2
确定被测试模型文件
例如:output/faster_rcnn_alt_opt/voc_2007_trainval/VGG16_faster_rcnn_final.caffemodel

3
确定配置文件
例如:experiments/cfgs/faster_rcnn_alt_opt.yml

4
确定测试集
例如:voc_2007_test

************

三 执行测试

1
训练完成后,删除data/VOCdevkit2007/annotations_cache/annots.pkl data/cache/voc_2007_trainval_gt_roidb.pkl,否则测试时报错。

2
faster rcnn根目录下打开终端,运行命令:
python tools/test_net.py --gpu 0 --def models/pascal_voc/VGG16/faster_rcnn_alt_opt/faster_rcnn_test.pt  --net output/faster_rcnn_alt_opt/voc_2007_trainval/VGG16_faster_rcnn_final.caffemodel --cfg experiments/cfgs/faster_rcnn_alt_opt.yml --imdb voc_2007_test

(提示:可将 --XXXX 等信息写入test_net.py中对应的default,减短命令长度)

************

四 查看测试结果

运行完test_net.py后,终端将显示出mAP值。
结果保存在output/faster_rcnn_alt_opt/voc_2007_test/VGG16_faster_rcnn_final/ 下。




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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值