您的位置:首页 > 编程语言 > MATLAB

使用matlab训练mnist模型

2016-12-07 17:14 609 查看
前面的博文http://blog.csdn.net/swj110119/article/details/53423957是通过命令进行mnist模型训练与测试的,由于实验需要,想要通过Matlab语句来实现mnist模型的训练,从而把这种方式用于其他问题模型的训练与测试。

1、准备数据与参数

因为matlab程序文件是在matlab/demo下,为了方便,直接把需要的文件拷贝到demo下:mnist_data文件(包含转化为lmdb的两个文件:mnist_train_lmdb、mnist_test_lmdb,均值文件mean.binaryproto和lent.prototxt),以及lenet_solver.prototxt和lenet_train_test.prototxt文件





2、修改网络参数:





3、然后在demo下新建mnist_train.m文件,程序如下:

%addpath('..')
caffe.reset_all
%读取网络参数
solver=caffe.Solver('lenet_solver.prototxt');
%训练模型
solver.solve();


这段程序需要运行的时间久一点,最终会在mnist_data下生成

4个文件



主要按照:http://blog.csdn.net/zb1165048017/article/details/52653501,博主很给力!研究了很久caffe了,看完这篇博文,有些东西之前不明白的、不知道该怎么做的,现在都有跟清楚的认识。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息