简单介绍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/ 下。
在Caffe中测试Faster R-CNN模型
最新推荐文章于 2024-01-12 19:48:37 发布