深度学习Caffe实战（9）Windows 平台caffe用MATLAB接口实现训练网络和测试

最新推荐文章于 2021-03-02 19:20:22 发布

原创

最新推荐文章于 2021-03-02 19:20:22 发布 · 9.6k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #matlab #windows #caffe

本文介绍了在Windows平台上，如何使用MATLAB接口进行Caffe的深度学习模型训练和测试。通过MATLAB脚本，详细展示了训练网络的过程，并提供了测试已训练模型的代码，包括数据预处理和结果可视化。内容涵盖了从创建脚本到运行模型的完整步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

上一篇介绍了网络协议中各个参数的作用，知道了各个参数的作用，想必应该可以尝试修改网络结构了。前几篇博客介绍的都是用命令行训练和测试网络，这篇博客介绍如何用MATLAB接口实现训练和测试网络，window平台下caffe使用者本来就少，这方面的资料更少了，下一篇博客介绍如何利用MATLAB提取特征和可视化feature map。话不多说，我们开始train

1、MATLAB实现卷积神经网络训练
打开MATLAB，进入affe-windows-master\matlab\demo\在这个文件夹下新建自己的文件夹（当然，不新建也行，后面注意路径就是了），我新建Heroin文件夹，在文件夹下建一个train文件夹，把数据和网络协议都拷贝到这个文件夹下，至于协议文件怎么修改，看我之前的博客写的很详细了，，，，
这里写图片描述

然后新建一个matlab脚本文件，内容如下：

clear;
clc;
addpath ../../../../matlab   %在不同的文件夹下，这个路径要注意一下
caffe.reset_all  
caffe.set_mode_cpu();%设置cpu或gpu模式
%caffe_model = 'snapshot/lenet_iter_10000.caffemodel'; %微调使用
solver=caffe.Solver('lenet_solver.prototxt'); %协议文件所在路径
%solver.net.copy_from(caffe_model);  %微调使用
% solver.solve(); %一次性训练
% loss=[];  
% accuracy=[];  

close all;
hold on%画图用的 
iter_ = solver.iter();%获取迭代次数
while iter_<10000
    solver.step(1);%一步一步迭代
    iter_ = solver.iter();    %得到迭代次数
    loss=solver.