前一篇博客介绍了如何利用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"
input_dim: 1
input_dim: 256
input_dim: 50
input_dim: 50
把roi_pool5层bottom修改为:
layer {
bottom: "data" #修改
bottom: "rois" #修改
top: "pool5"
name: "roi_pool5"
type: "ROIPooling"
roi_pooling_param {
pooled_w: 6
pooled_h: 6
spatial_scale: 0.0625 # (1/16)
}
}
3、运行script_faster_rcnn_demo.m得到结果
虽然我为了跑流程,标注的样本很少,但是结果还不错。