Caffe_Windows学习笔记(五)用训练好的caffemodel来进行分类

0、参考文献

[1]denny http://www.cnblogs.com/denny402/p/5111018.html

1、准备工作

1)下载一张波斯猫的图片,存放在CAFFE_ROOT/examples/images/jfm.jpg

2)caffemodel文件

开发caffe的贾大牛团队,利用imagenet图片和caffenet模型训练好了一个caffemodel,  供大家下载。要进行图片的分类,这个caffemodel是最好不过的了。

下载地址为:http://dl.caffe.berkeleyvision.org/bvlc_reference_caffenet.caffemodel

3)均值文件

有了caffemodel文件,就需要对应的均值文件,在测试阶段,需要把测试数据减去均值。这个文件我们用脚本来下载,用cygdrive在caffe根目录下执行:

./data/ilsvrc12/get_ilsvrc_aux.sh

执行并下载后,均值文件放在 data/ilsvrc12/ 文件夹里,imagenet_mean.binaryproto。

4)synset_words.txt文件

在调用脚本文件下载均值的时候,这个文件也一并下载好了。里面放的是1000个类的名称。

2、C++方法

在caffe根目录下的 examples/cpp-classification/ 文件夹下面,有个classification.cpp文件,就是用来分类的。当然编译后,放在/Build/x64/Release/下面。

在cygdrive运行命令:

./Build/x64/Release/classification.exe \
  models/bvlc_reference_caffenet/deploy.prototxt \
  models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel \
  data/ilsvrc12/imagenet_mean.binaryproto \
  data/ilsvrc12/synset_words.txt \
  examples/images/jfm.jpg

运行成功后,输出top-5结果:


即有0.4160的概率为波斯猫, 有0.2546的概率为Samoyed ......


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值