验证自己的caffe是否配置成功

14 篇文章 0 订阅

花了好几天的时间对电脑进行了caffe的配置,真的caffe真的配置太繁琐了,特别的麻烦。现在呢,就是验证一下是否是配置成功。

我的caffe的配置是对ubuntu自带的python2.7版本下的caffe配置。

网上有说直接在终端中执行 caffe命令如果后面有提示命令就算成功,但是我的不是这样,如下。一时间把我搞得心烦意乱。

然后,有说是在python下导入caffe如果不报错,那就是成功,我试试下,这次成功了,如下图。我现在真的是要疯掉了,到底是谁的说法是正确的,我自己都不知道了。

为了正进一步的验证,我准备运行caffe自带的mnist示例,来验证

1、先进入caffe文件目录,再用data/mnist下的get_mnist.sh下载MNIST数据集,代码如下:

我的caffe文件路径是/home/zhai/experiment/caffe

执行

sudo ./data/mnist/get_mnist.sh

这里我的电脑下载非常非常非长的慢,我尝试着去解决,但是就是不行,然后我就从get_mnist.sh 中找到链接地址,在window下下载后,放到caffe/data/mnist这个文件夹下。

会得到四个.gz文件,然后提取到本文件夹下,如下图的红框(注意:提取后的文件命名一定要是对应的,如下文件夾中有一个没有框出来就是因为,他的文件名字中间是 . 而不是 - )

 2、转换成lmdb格式

还是在当前根目录caffe下,执行

./examples/mnist/create_mnist.sh

查看文件

具体的数据内容格式为

先介绍下

训练网络:

网络结构定义在./examples/mnist/lenet_train_test.prototxt中。

训练参数配置在./examples/mnist/lenet_solver.prototxt中。

3、训练

通过命令行执行训练,在根目录caffe下,执行代码如下:

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

正确率:99.15%(使用GPU)

实验输出模型

 4、测试

./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel

 注意别输错命令了。

 结果

至此,证明我的基于python2.7版本的caffe配置成功。

 

参考:

https://blog.csdn.net/weixin_41770169/article/details/90370123

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值