使用caffe和mnsit数据集训练生成的lenet_iter_10000.caffemodel的使用方法

本文面向初学者,介绍了如何在Caffe中使用lenet_iter_10000.caffemodel对MNIST数据集进行分类。步骤包括找到classification.bin,确定deploy.prototxt、.caffemodel、mean.binaryproto、labels.txt和测试图片img.jpg等必要文件,并解释了它们的作用。文章提供了创建mean.binaryproto和labels.txt的详细过程,以及运行分类脚本的命令。
摘要由CSDN通过智能技术生成

醒目:本文只面向初入caffe的新手

在成功配置好caffe后,一般我们会使用mnist数据集进行训练,以检测caffe是否正常安装。caffe的官网提供了使用mnist数据集进行lenet神经网络训练的方法。见Training LeNet on MNIST with Caffe,按该教程的引导,我们最后训练得到了lenet_iter_6000.caffemodel和lenet_iter_10000.caffemodel两个文件。但是Caffe官网的教程并没有给出这两个文件怎么使用,只是简单地简单提了一句“you can deploy as a trained model in your application”,导致很多像我这样的新手到这里就摸不到头脑,难以继续下去。下面分享一下这个文件的使用方法,并用这个model来识别图片中的数字:

第一步:找到分类脚本classification.bin

该文件位于{caffe}/build/examples/cpp_classification目录下,其中{caffe}是你caffe的安装目录。现在我们试着运行一下它,看它提示我们需要哪些参数:

[sevend@sevend-pc caffe]$ ./build/examples/cpp_classification/classification.bin 
Usage: ./build/examples/cpp_classification/classification.bin deploy.prototxt network.caffemodel mean.binaryproto labels.txt img.jpg

可以看到,它提示我们,运行这个脚本需要五个参数,分别是:

deploy.prototxt
network.caffemodel
mean.binaryproto
labels.txt
img.jpg

那么,接下来我们的目标就是找出这五个参数该传什么进去。在本文,我们不细究这五个参数到底是干什么的(那需要非常多的篇幅,本文只面向初入caffe的新手),只告诉大家成功跑起来的方法。在这里告诉大家,img.jpg是要识别的图片,大家可以去网上找一些手写数字图片,或者自己写一个(文末提供了我所用的测试图片)。接下来我们就挨个找到这五个文件:

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值