caffe(ubuntu14.04)学习笔记1——运行MNIST数据集模型

这篇博客介绍了如何在Ubuntu14.04上使用Caffe框架处理MNIST数据集。首先,详细说明了MNIST数据集的基本信息和下载步骤,接着讲述了如何将原始数据转换为Caffe可读的leveldb或lmdb格式。最后,展示了在CPU和GPU上使用LeNet-5模型进行训练的时间和性能结果,以及CPU训练的精度和损失值。
摘要由CSDN通过智能技术生成

MNIST数据集简介:

        MNIST数据集是一个大型的手写体数据库,广泛用于机器学习领域的训练和测试,它是由纽约大学的Yann LeCun教授整理的,包括60000个训练样本和10000个测试样本,其图像都是灰度图像,固定尺寸为28x28(像素)。

数据集的准备:

caffe源码框架的data/mnist文件夹下有MNIST数据集的下载脚本get_mnist.sh.

1.首先将路径切换到caffe的根目录下


2.进行原数据集的下载

(首先先检查你的网络是否连接哦)

运行get_mnist.sh脚本


(运行的图片知识截取了其中的一部分),运行完成后在data/mnist文件夹下会出现以下四个二进制文件(根据自己的情况,如果下载下来是压缩包则需要进行解压)。


caffe可以识别的数据文件为LEVELDB和LMDB文件,因此运行一下脚本将二进制文件转换为leveldb或者lmdb文件(这两种格式的数据可以加快系统加载数据的速度)。


即可在examples/mnist文件夹下发现这两个文件


此时数据准备工作完成。

LeNet-5模型的训练:

运行如下命令进行模型的训练


训练结果:(因CPU或GPU的性能不同而异)

在CPU上训练:

用时:           21分43秒;

分类准确率:99.21%;

loss:             0.0286289


在GPU上训练:

用时:            2分54秒

分类准确率:99.09%

loss:             0.0294631

附一张CPU上的训练结果图~


如有错误,欢迎大家指正~~





评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值