码迷,mamicode.com
首页 > 其他好文 > 详细

将caffe训练时loss的变化曲线用matlab绘制出来

时间:2016-04-23 21:10:48      阅读:196      评论:0      收藏:0      [点我收藏+]

标签:

1. 首先是提取 训练日志文件;

 

2. 然后是matlab代码:

clear all; close all; clc;
log_file = /home/wangxiao/Downloads/43_attribute_baseline.log;
fid = fopen(log_file, r);
fid_accuracy = fopen(/home/wangxiao/Downloads/output_accuracy.txt, w); 
fid_loss = fopen(/home/wangxiao/Downloads/output_loss.txt, w);

iteration ={};
loss = {};
accuracy = {};
path = /home/wangxiao/Downloads/;
fid_ = fopen([path, loss_file_.txt], a);
while(~feof(fid))
    tline = fgetl(fid);
    %%
    if strfind(tline, sgd_solver.cpp:)
        iter_index = strfind(tline, Iteration );
        rest = tline((iter_index+9):end);
        iter_current = strtok(rest, ,);                   % iteration number;
        iteration = [iteration  iter_current];        % count the iteration; 
        lr_index = strfind(tline, lr = );
        lr_current = tline((lr_index+4):end);                  % learning rate;
    end
    
    %%
    if strfind(tline, solver.cpp:228])
        iter_index = strfind(tline, loss = );
        loss_current = tline((iter_index+7):end);
        fprintf(fid_, %s \n, loss_current );
        loss = [loss  loss_current] ;       % count the iteration; 
    end
    
    if strfind(tline, aver_accuracy: )
        aver_accuracy_index = strfind(tline, aver_accuracy: );
        aver_accuracy_current = tline((aver_accuracy_index+15):end);
       
        accuracy = [accuracy  aver_accuracy_current];
    end
end
    

loss_file_Path = importdata(/home/wangxiao/Downloads/loss_file_.txt);



 plot(loss_file_Path)

 

3. 结果展示:

技术分享

将caffe训练时loss的变化曲线用matlab绘制出来

标签:

原文地址:http://www.cnblogs.com/wangxiaocvpr/p/5425427.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!