Win10下用“命令行+Matlab”实现Caffe训练过程可视化

1.训练模型并获取训练日志

用CMD调用Caffe进行模型训练并将日志重定向到txt文件,命令格式如下:

#命令格式:
command 2> logfile_path & type logfile_path

#示例:
Build\x64\Release\caffe.exe  train --solver=examples\mnist\lenet_solver_adam.prototxt  2>  examples\mnist\train_log.txt  &  type  examples\mnist\train_log.txt
#回车执行命令,训练完成后,CMD窗口的训练日志会被保存到examples\mnist\train_log.txt中。
2.解析日志文件并绘制loss和accuracy曲线

将上一步得到的日志文件train_log.txt的完整路径输入下面Matlab脚本定义的PlotLossAndAccuracy()函数,即可解析并绘制loss和accuracy曲线

function result=PlotLossAndAccuracy(kLogName)
%输入caffe模型训练日志的路径,绘制loss和accuracy曲线
%返回值result=[accuracy,testloss,trainloss]
%示例:
%PlotLossAndAccuracy('E:\caffe-window\新建文本文档.txt')

%初始化变量
ind=strfind(kLogName,'\');%找到所有'\'
ind1=ind(end)+1;
ind=strfind(kLogName,'_train_log');
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值