DL开源框架Caffe | 用训练好的模型对数据进行预测

一句话理解Caffe:

Caffe的万丈高楼(Net)是按照我们设计的图纸(prototxt),用很多砖块(Blob)筑成一层层(Layer)楼房,最后通过某些手段(Solver)进行简装修(Train)/精装修(Finetune)实现的,另外每个楼层都可以被购买规己用,那么就会有归属(Lable)。

一 Caffe识别问题上利用训练好的模型预测

利用已有的模型可以对测试数据集进行预测,命令:

./build/tools/caffe.bin test \
> -model examples/mnist/lenet_train_test.prototxt \
> -weights examples/mnist/lenet_iter_10000.caffemodel \
> -iterations 100

命令行解析:

./build/tools/caffe.bin test \
//表示只做预测,不进行参数更新
> -model examples/mnist/lenet_train_test.prototxt \
//指定模型描述文本文件
> -weights examples/mnist/lenet_iter_10000.caffemodel \
//指定模型预先训练好的权值文件
> -iterations 100
//指定测试迭代次数。参与测试样例数目 = 迭代次数 * batch_size

输出日志:

二 Caffe检测问题上利用训练好的模型预测

这里主要针对py-faster-rcnn的目标检测模型来讲,训练完成的model如何直接用来测试自己的图像呢?

1、训练完成之后,将output/faster_rcnn_end_to_end/voc_2007_trainval中的最终模型拷贝到data/faster_rcnn_models(删除以前生成类似的model)中。

2、修改/tools/demo.py为(最好拷贝一份修改):

(1) CLASSES =(‘background‘, ‘xxx’)(这里是你的类别名)

(2) NETS ={‘vgg16’: (‘VGG16’, ‘VGG16_faster_rcnn_final.caffemodel’),       ‘zf’:  (‘ZF’,’ZF_faster_rcnn_final.caffemodel’)} (这里是你训练完成的model)

(3) 在训练集图片中找一张出来放入py-faster-rcnn/data/demo文件夹中,命名为000001.jpg。

  im_names = ['000001.jpg'](需要测试的图像名称)

(4) 根据你自己的训练方式修改命令行输入参数,默认的训练方式为alt_opt的!

3、运行demo,即在py-faster-rcnn文件夹下终端输入: ./tools/demo.py –net zf

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏北京马哥教育

实战Google深度学习框架:TensorFlow计算加速

作者:才云科技Caicloud,郑泽宇,顾思宇 要将深度学习应用到实际问题中,一个非常大的问题在于训练深度学习模型需要的计算量太大。比如Inception-v3...

33670
来自专栏mathor

“达观杯”文本智能处理挑战赛

 由于提供的数据集较大,一般运行时间再10到15分钟之间,基础电脑配置在4核8G的样子(越消耗内存在6.2G),因此,一般可能会遇到内存溢出的错误

39820
来自专栏人人都是极客

5.训练模型之利用训练的模型识别物体

接下来我们开始训练,这里要做三件事: 将训练数据上传到训练服务器,开始训练。 将训练过程可视化。 导出训练结果导出为可用作推导的模型文件。 配置 Pipelin...

42440
来自专栏人工智能LeadAI

YOLO:实时目标检测

一瞥(You Only Look Once, YOLO),是检测Pascal VOC(http://host.robots.ox.ac.uk:8080/pasc...

1.1K70
来自专栏专知

Tensorflow实战系列:手把手教你使用CNN进行图像分类(附完整代码)

【导读】专知小组计划近期推出Tensorflow实战系列,计划教大家手把手实战各项子任务。本教程旨在手把手教大家使用Tensorflow构建卷积神经网络(CNN...

4.7K40
来自专栏CNN

MobileNet V1官方预训练模型的使用

MobileNet V1的网络结构可以直接从官方Github库中下载定义网络结构的文件,地址为:https://raw.githubusercontent.co...

77520
来自专栏数据小魔方

时间管理的工具——甘特图(Gantt chart)

今天跟大家分享一种用作时间管理的工具——甘特图(Gantt Chart)。 ▽▼▽ 这种图表的制作理念非常简单,就是通过设定项目开始时间和持续时间,利用堆积条形...

74570
来自专栏计算机视觉与深度学习基础

【深度学习】使用tensorflow实现AlexNet

AlexNet是2012年ImageNet比赛的冠军,虽然过去了很长时间,但是作为深度学习中的经典模型,AlexNet不但有助于我们理解其中所使用的很多技巧,...

477100
来自专栏素质云笔记

SSD+caffe︱Single Shot MultiBox Detector 目标检测+fine-tuning(二)

承接上一篇SSD介绍:SSD+caffe︱Single Shot MultiBox Detector 目标检测(一) 如果自己要训练SSD模型呢,关键...

1.1K100
来自专栏AI研习社

Github 项目推荐 | TensorFlow 概率推理工具集 —— probability

Probability 是 TensorFlow 的概率推理工具集,它是集建模工具、推理算法、一些有用的模型和一般统计计算于一身的开发工具集合。利用 Tenso...

59640

扫码关注云+社区

领取腾讯云代金券