在Caffe中测试Faster R-CNN模型

来源：互联网发布：潍坊山河软件编辑：程序博客网时间：2024/06/05 16:22

简单介绍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 0