前一篇博客介绍了如何利用Faster-RCNN训练自己的数据集,训练好会得到一个模型,这篇博客介绍如何利用训练好的模型进行测试数据。
1、训练好的模型存放位置
训练好的模型存放在faster_rcnn-master\output\faster_rcnn_final\faster_rcnn_VOC2007_ZF,把script_faster_rcnn_demo.m文件拷贝到faster_rcnn-master文件夹下,然后把模型路径改成训练好的模型路径
model_dir = fullfile(pwd, 'output', 'faster_rcnn_final', 'faster_rcnn_VOC0712_ZF');
2、修改协议文件
打开faster_rcnn-master\output\faster_rcnn_final\faster_rcnn_VOC2007_ZF\detection_test.prototxt
把relu5包括relu5以前的层都删除,
把input data修改为如下:
input: "data"
inpu