win7 caffe使用笔记——绘制学习曲线

最新推荐文章于 2024-01-22 08:30:06 发布

Gotta-C

最新推荐文章于 2024-01-22 08:30:06 发布

阅读量1k

点赞数

分类专栏： caffe

本文链接：https://blog.csdn.net/cgt19910923/article/details/55271431

版权

caffe 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

1.命令控制台输出重定向，保存训练的日志文件。

train.bat

G:
cd G:\caffe-master
set GLOG_logtostderr=1
set BIN=Build/x64/Release

"%BIN%/caffe.exe" train --solver=examples/Planthopper_test/Planthopper_solver-leveldb.prototxt 

>examples/Planthopper_test/Log/net.log 2>&1
pause

2.筛选日志文件中的loss，accuracy，迭代行，并绘制曲线。

drawcurve.m

% Well, this is a function that write the
% iteration vs accurancy
% iteration vs loss
% To a file

clc;
clear;

% log file of caffe model
logName = 'net.log';

fid = fopen(logName, 'r');
fid_accuracy = fopen('output_accuracy.txt', 'w');
fid_test_loss = fopen('output_test_loss.txt','w'); 
fid_train_loss = fopen('output_train_loss.txt', 'w');

tline = fgetl(fid);
accur=zeros(401,1);      %max_iteration/test_interval  精度,测试损失数量
testloss=zeros(401,1);
m=1;
trainloss=zeros(2001,1);  %max_iteration/display       训练损失数量
n=1;
while ischar(tline)    %读入每一行
    % First find the accuracy line
    k = strfind(tline, 'Test net output');
    if (k)
        k = strfind(tline, 'accuracy');
        if (k)
            % If the string contain test and accuracy at the same time
            % The bias from 'accuracy' to the float number
            indexStart = k + 11; %定位到精度数值
            indexEnd = size(tline);
            str = tline(indexStart : indexEnd(2));  %读入精度
        
        fprintf(fid_accuracy, '%s\r\n', str);
        accur(m)=str2double(str);
        end
    end
  % next,find the test loss line
      k5 = strfind(tline, 'Test net output');
    if (k5)
        k5 = strfind(tline, 'loss');
        if (k5)
            indexStart = k5 + 7; %定位到test loss数值
            indexEnd = indexStart + 6;
            str5 = tline(indexStart : indexEnd);  %读入test loss 
        fprintf(fid_test_loss, '%s\r\n', str5);
      testloss(m)=str2double(str5);
      m=m+1;
        end
    end  
    
    % Then find the loss line
    k1 = strfind(tline, 'Iteration');
    if (k1)
       k2 = strfind(tline, 'loss');
       if (k2)
           indexStart = k2 + 7;        %定位到train loss数值
           indexEnd = size(tline);
           str1 = tline(indexStart:indexEnd(2));        %取train loss
           indexStart = k1 + 10;                     %定位到iteration
           indexEnd = strfind(tline, ',') - 1;    
           str2 = tline(indexStart:indexEnd);        %取迭代序号
           res_str1 = strcat(str2, '/', str1);
           fprintf(fid_train_loss, '%s\r\n', res_str1);
           trainloss(n)=str2double(str1);
           n=n+1;
       end
    end
    tline = fgetl(fid);
end

fclose(fid);
fclose(fid_accuracy);

testidx = 0:400;
idx_test = testidx * 500;
trainidx = 0:2000;
idx_train = trainidx * 100;
figure;plot(idx_test,accur);
grid on;
legend('test accuracy');
xlabel('iterations');
ylabel('accuracy');

figure;
plot(idx_test,testloss);
hold on;
plot(idx_train,trainloss);
grid on;
legend('test loss', 'train loss');
xlabel('iterations');
ylabel('loss');

输出3个txt文件，train loss，test loss，accuracy。

绘制loss曲线和accuracy曲线。