您的位置:首页 > Web前端

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

2017-05-21 19:32 986 查看
前一篇博客介绍了如何利用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得到结果







虽然我为了跑流程,标注的样本很少,但是结果还不错。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐