深度学习Caffe实战笔记(21)Windows平台 Faster-RCNN 训练好的模型测试数据

前一篇博客介绍了如何利用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得到结果
深度学习Caffe实战笔记(21)Windows平台 Faster-RCNN 训练好的模型测试数据_第1张图片
深度学习Caffe实战笔记(21)Windows平台 Faster-RCNN 训练好的模型测试数据_第2张图片
深度学习Caffe实战笔记(21)Windows平台 Faster-RCNN 训练好的模型测试数据_第3张图片

虽然我为了跑流程,标注的样本很少,但是结果还不错。

你可能感兴趣的:(caffe-深度学习)